Searched refs:is_batch (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/python/ops/ |
| D | image_ops_impl.py | 696 is_batch = True
699 is_batch = False
702 is_batch = False
739 if not is_batch:
779 is_batch = True
782 is_batch = False
785 is_batch = False
819 if not is_batch:
857 is_batch = True
859 is_batch = False
[all …]
|
| D | image_ops_test.py | 155 is_batch = True
157 is_batch = False
169 if not is_batch:
|
| /external/tensorflow/tensorflow/cc/gradients/ |
| D | math_grad_test.cc | 575 void TestMatMulGrad(const bool is_batch, const bool t_x, const bool t_y) { in TestMatMulGrad() argument
579 RandMatMulShapes(is_batch, t_x, t_y, &shapes); in TestMatMulGrad()
588 if (is_batch) { in TestMatMulGrad()
600 void RandMatMulShapes(const bool is_batch, const bool tx, const bool ty, in RandMatMulShapes() argument
610 if (is_batch) { in RandMatMulShapes()
620 if (is_batch) { in RandMatMulShapes()
630 if (is_batch) { in RandMatMulShapes()
|
| D | math_grad.cc | 1047 Status MatMulGradHelper(const Scope& scope, const bool is_batch, in MatMulGradHelper() argument
1052 if (is_batch == false) { in MatMulGradHelper()
1074 const bool is_batch, in MatMulGradCommon() argument
1081 if (is_batch == false) { in MatMulGradCommon()
1093 return MatMulGradHelper(scope, is_batch, grad_inputs[0], false, b, true, a, in MatMulGradCommon()
1096 return MatMulGradHelper(scope, is_batch, grad_inputs[0], false, b, false, in MatMulGradCommon()
1099 return MatMulGradHelper(scope, is_batch, b, false, grad_inputs[0], true, a, in MatMulGradCommon()
1102 return MatMulGradHelper(scope, is_batch, b, true, grad_inputs[0], true, in MatMulGradCommon()
|
| /external/libcxx/utils/google-benchmark/include/benchmark/ |
| D | benchmark.h | 677 bool KeepRunningInternal(size_t n, bool is_batch);
694 bool is_batch) { in KeepRunningInternal() argument
699 assert(is_batch || n == 1); in KeepRunningInternal()
712 if (is_batch && total_iterations_ != 0) { in KeepRunningInternal()
|
| /external/google-benchmark/include/benchmark/ |
| D | benchmark.h | 677 bool KeepRunningInternal(size_t n, bool is_batch);
694 bool is_batch) { in KeepRunningInternal() argument
699 assert(is_batch || n == 1); in KeepRunningInternal()
712 if (is_batch && total_iterations_ != 0) { in KeepRunningInternal()
|