Searched defs:outer_dim (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | reverse_op_test.cc | 201 int outer_dim, int middle_dim, in RunReverseRowsBenchmark()
215 const int outer_dim = state.range(0); in BM_ReverseRowsOf1Channel_1T_float() local
229 const int outer_dim = state.range(0); in BM_ReverseRowsOf1Channel_1T_uint8() local
243 const int outer_dim = state.range(0); in BM_ReverseRowsOf1Channel_4T_float() local
257 const int outer_dim = state.range(0); in BM_ReverseRowsOf1Channel_4T_uint8() local
271 const int outer_dim = state.range(0); in BM_ReverseRowsOf3Channels_1T_float() local
286 const int outer_dim = state.range(0); in BM_ReverseRowsOf3Channels_1T_uint8() local
301 const int outer_dim = state.range(0); in BM_ReverseRowsOf3Channels_4T_float() local
316 const int outer_dim = state.range(0); in BM_ReverseRowsOf3Channels_4T_uint8() local
330 const int outer_dim = state.range(0); in BM_ReverseRowsOf4Channels_1T_float() local
[all …]
|
| D | redux_functor.h | 51 Eigen::Index inner_dim = 1, outer_dim = 1; in operator() local
220 Eigen::Index inner_dim = 1, middle_dim = 1, outer_dim = 1; in operator() local
|
| D | reverse_op.cc | 60 for (int outer_dim = start; outer_dim < end; ++outer_dim) { in ReverseRows() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | dot_merger.cc | 167 int64_t outer_dim = contracting_dim == dnums.lhs_batch_dimensions_size() in TryMergeSameOperand() local
|
| /external/tensorflow/tensorflow/python/kernel_tests/math_ops/ |
| D | segment_reduction_ops_test.py | 1098 def _runGraph(self, op_functor, outer_dim, ratio, inner_dim, dtype): argument
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | reference_ops.h | 922 int outer_dim = std::min(batch_dim, seq_dim); in ReverseSequence() local
|