Home
last modified time | relevance | path

Searched refs: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() argument
201 TensorShape shape{outer_dim, middle_dim, channels}; in RunReverseRowsBenchmark()
209 static void BM_ReverseRowsOf1Channel_1T_float(int iters, int outer_dim, in BM_ReverseRowsOf1Channel_1T_float() argument
211 RunReverseRowsBenchmark<float>(iters, outer_dim, middle_dim, in BM_ReverseRowsOf1Channel_1T_float()
220 static void BM_ReverseRowsOf1Channel_1T_uint8(int iters, int outer_dim, in BM_ReverseRowsOf1Channel_1T_uint8() argument
222 RunReverseRowsBenchmark<uint8>(iters, outer_dim, middle_dim, in BM_ReverseRowsOf1Channel_1T_uint8()
231 static void BM_ReverseRowsOf1Channel_4T_float(int iters, int outer_dim, in BM_ReverseRowsOf1Channel_4T_float() argument
233 RunReverseRowsBenchmark<float>(iters, outer_dim, middle_dim, in BM_ReverseRowsOf1Channel_4T_float()
242 static void BM_ReverseRowsOf1Channel_4T_uint8(int iters, int outer_dim, in BM_ReverseRowsOf1Channel_4T_uint8() argument
244 RunReverseRowsBenchmark<uint8>(iters, outer_dim, middle_dim, in BM_ReverseRowsOf1Channel_4T_uint8()
[all …]
Dredux_functor.h44 int64 outer_dim = 1; in operator() local
45 for (int i = 0; i < num_dims - 1; ++i) outer_dim *= input_dims[i]; in operator()
49 const int64 total_workload = outer_dim * inner_dim; in operator()
63 const int64 outer_block_size = Eigen::divup(outer_dim, num_blocks); in operator()
81 input_data, outer_dim](Eigen::Index start, in operator()
86 outer_dim_limit = std::min(outer_dim, outer_dim_limit); in operator()
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
928 output_outer_dim = int(outer_dim / ratio)
929 const = np.random.randint(5, size=(outer_dim, inner_dim))
930 seg_ids = np.sort(np.random.randint(output_outer_dim, size=outer_dim))
943 [name, outer_dim, ratio, inner_dim,
950 for outer_dim, ratio, inner_dim, dtype in itertools.product(*self.options):
953 self._runGraph(op_functor, outer_dim, ratio, inner_dim, dtype)
958 for outer_dim, ratio, inner_dim, dtype in itertools.product(*self.options):
961 self._runGraph(op_functor, outer_dim, ratio, inner_dim, dtype)
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dreference_ops.h4692 int outer_dim = std::min(batch_dim, seq_dim); in ReverseSequence() local
4694 for (int i = 0; i < outer_dim; ++i) { in ReverseSequence()
4699 for (int i = outer_dim + 1; i < medium_dim; ++i) { in ReverseSequence()
4708 const int dims_at_outer_dim = input_shape.Dims(outer_dim); in ReverseSequence()