Home
last modified time | relevance | path

Searched refs: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() argument
204 TensorShape shape{outer_dim, middle_dim, channels}; in RunReverseRowsBenchmark()
215 const int outer_dim = state.range(0); in BM_ReverseRowsOf1Channel_1T_float() local
218 RunReverseRowsBenchmark<float>(state, outer_dim, middle_dim, in BM_ReverseRowsOf1Channel_1T_float()
229 const int outer_dim = state.range(0); in BM_ReverseRowsOf1Channel_1T_uint8() local
232 RunReverseRowsBenchmark<uint8>(state, outer_dim, middle_dim, in BM_ReverseRowsOf1Channel_1T_uint8()
243 const int outer_dim = state.range(0); in BM_ReverseRowsOf1Channel_4T_float() local
246 RunReverseRowsBenchmark<float>(state, outer_dim, middle_dim, in BM_ReverseRowsOf1Channel_4T_float()
257 const int outer_dim = state.range(0); in BM_ReverseRowsOf1Channel_4T_uint8() local
260 RunReverseRowsBenchmark<uint8>(state, outer_dim, middle_dim, in BM_ReverseRowsOf1Channel_4T_uint8()
[all …]
Dredux_functor.h51 Eigen::Index inner_dim = 1, outer_dim = 1; in operator() local
53 outer_dim *= input_dims[i]; in operator()
57 if (1 == outer_dim) { in operator()
93 const auto compute = [inner_dim, outer_dim, num_blocks, inner_block_size, in operator()
105 for (Eigen::Index i = 0; i < outer_dim; ++i) { in operator()
114 const Eigen::Index compute_size = outer_dim * inner_block_size; in operator()
129 const Eigen::Index total_workload = outer_dim * inner_dim; in operator()
142 const Eigen::Index outer_block_size = Eigen::divup(outer_dim, num_blocks); in operator()
160 buffer_data, input_data, outer_dim]( in operator()
165 outer_dim_limit = std::min(outer_dim, outer_dim_limit); in operator()
[all …]
Dreverse_op.cc60 for (int outer_dim = start; outer_dim < end; ++outer_dim) { in ReverseRows() local
/external/tensorflow/tensorflow/python/kernel_tests/
Dsegment_reduction_ops_test.py975 def _runGraph(self, op_functor, outer_dim, ratio, inner_dim, dtype): argument
976 output_outer_dim = int(outer_dim / ratio)
977 const = np.random.randint(5, size=(outer_dim, inner_dim))
978 seg_ids = np.sort(np.random.randint(output_outer_dim, size=outer_dim))
991 [name, outer_dim, ratio, inner_dim,
998 for outer_dim, ratio, inner_dim, dtype in itertools.product(*self.options):
1001 self._runGraph(op_functor, outer_dim, ratio, inner_dim, dtype)
1006 for outer_dim, ratio, inner_dim, dtype in itertools.product(*self.options):
1009 self._runGraph(op_functor, outer_dim, ratio, inner_dim, dtype)
/external/tensorflow/tensorflow/python/ops/ragged/
Dragged_tensor.py2358 outer_dim = tensor_shape.dimension_at_index(self._shape, 0)
2359 outer_splits_shape = [None if outer_dim is None else outer_dim + 1]
2446 outer_dim = tensor_shape.dimension_value(self._shape[0])
2447 if outer_dim is not None:
2448 result.row_splits.set_shape([outer_dim + 1])
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dreference_ops.h1727 int outer_dim = std::min(batch_dim, seq_dim); in ReverseSequence() local
1729 for (int i = 0; i < outer_dim; ++i) { in ReverseSequence()
1734 for (int i = outer_dim + 1; i < medium_dim; ++i) { in ReverseSequence()
1743 const int dims_at_outer_dim = input_shape.Dims(outer_dim); in ReverseSequence()