Searched refs:is_batch (Results 1 - 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/python/ops/ |
| H A D | image_ops_impl.py | 582 is_batch = True
585 is_batch = False
588 is_batch = False
625 if not is_batch:
665 is_batch = True
668 is_batch = False
671 is_batch = False
705 if not is_batch:
743 is_batch = True
745 is_batch
[all...] |
| H A D | image_ops_test.py | 149 is_batch = True
151 is_batch = False
163 if not is_batch:
|
| /external/tensorflow/tensorflow/cc/gradients/ |
| H A D | math_grad_test.cc | 577 void TestMatMulGrad(const bool is_batch, const bool t_x, const bool t_y) { argument
581 RandMatMulShapes(is_batch, t_x, t_y, &shapes);
590 if (is_batch) {
602 void RandMatMulShapes(const bool is_batch, const bool tx, const bool ty, argument
612 if (is_batch) {
622 if (is_batch) {
632 if (is_batch) {
|
| H A D | math_grad.cc | 1011 Status MatMulGradHelper(const Scope& scope, const bool is_batch, argument
1016 if (is_batch == false) {
1038 const bool is_batch,
1045 if (is_batch == false) {
1057 return MatMulGradHelper(scope, is_batch, grad_inputs[0], false, b, true, a,
1060 return MatMulGradHelper(scope, is_batch, grad_inputs[0], false, b, false,
1063 return MatMulGradHelper(scope, is_batch, b, false, grad_inputs[0], true, a,
1066 return MatMulGradHelper(scope, is_batch, b, true, grad_inputs[0], true,
1037 MatMulGradCommon(const Scope& scope, const Operation& op, const bool is_batch, const std::vector<Output>& grad_inputs, const string& attr_adj_x, const string& attr_adj_y, std::vector<Output>* grad_outputs) argument
|
Completed in 356 milliseconds