Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Dreverse_op_test.cc198 static void RunReverseRowsBenchmark(int iters, int outer_dim, int middle_dim, in RunReverseRowsBenchmark()
209 static void BM_ReverseRowsOf1Channel_1T_float(int iters, int outer_dim, in BM_ReverseRowsOf1Channel_1T_float()
220 static void BM_ReverseRowsOf1Channel_1T_uint8(int iters, int outer_dim, in BM_ReverseRowsOf1Channel_1T_uint8()
231 static void BM_ReverseRowsOf1Channel_4T_float(int iters, int outer_dim, in BM_ReverseRowsOf1Channel_4T_float()
242 static void BM_ReverseRowsOf1Channel_4T_uint8(int iters, int outer_dim, in BM_ReverseRowsOf1Channel_4T_uint8()
253 static void BM_ReverseRowsOf3Channels_1T_float(int iters, int outer_dim, in BM_ReverseRowsOf3Channels_1T_float()
265 static void BM_ReverseRowsOf3Channels_1T_uint8(int iters, int outer_dim, in BM_ReverseRowsOf3Channels_1T_uint8()
277 static void BM_ReverseRowsOf3Channels_4T_float(int iters, int outer_dim, in BM_ReverseRowsOf3Channels_4T_float()
289 static void BM_ReverseRowsOf3Channels_4T_uint8(int iters, int outer_dim, in BM_ReverseRowsOf3Channels_4T_uint8()
300 static void BM_ReverseRowsOf4Channels_1T_float(int iters, int outer_dim, in BM_ReverseRowsOf4Channels_1T_float()
[all …]
Dredux_functor.h44 int64 outer_dim = 1; in operator() local
Dreverse_op.cc63 for (int outer_dim = start; outer_dim < end; ++outer_dim) { in ReverseRows() local
/external/tensorflow/tensorflow/python/kernel_tests/
Dsegment_reduction_ops_test.py927 def _runGraph(self, op_functor, outer_dim, ratio, inner_dim, dtype): argument
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dreference_ops.h4692 int outer_dim = std::min(batch_dim, seq_dim); in ReverseSequence() local