Home
last modified time | relevance | path

Searched defs:outer_dim (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dreverse_op_test.cc201 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 …]
Dredux_functor.h51 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
Dreverse_op.cc60 for (int outer_dim = start; outer_dim < end; ++outer_dim) { in ReverseRows() local
/external/tensorflow/tensorflow/compiler/xla/service/
Ddot_merger.cc167 int64_t outer_dim = contracting_dim == dnums.lhs_batch_dimensions_size() in TryMergeSameOperand() local
/external/tensorflow/tensorflow/python/kernel_tests/math_ops/
Dsegment_reduction_ops_test.py1098 def _runGraph(self, op_functor, outer_dim, ratio, inner_dim, dtype): argument
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dreference_ops.h922 int outer_dim = std::min(batch_dim, seq_dim); in ReverseSequence() local