Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A 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 …]
H A 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
H A Dreverse_op.cc60 for (int outer_dim = start; outer_dim < end; ++outer_dim) { in ReverseRows() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DIndexing.cu52 int64_t numel, int64_t stride, int64_t stride_before, int64_t outer_dim, bool accumulate) { in indexing_backward_kernel()
130 int64_t numel, int64_t stride, int64_t stride_before, int64_t outer_dim, bool accumulate) { in indexing_backward_kernel_stride_1()
186 int64_t numel, int64_t stride, int64_t stride_before, int64_t outer_dim, bool accumulate) { in indexing_backward_kernel_small_stride()
230 int64_t numel, int64_t stride, int64_t stride_before, int64_t outer_dim, in indexing_backward_kernel_quantized()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Ddot_merger.cc167 int64_t outer_dim = contracting_dim == dnums.lhs_batch_dimensions_size() in TryMergeSameOperand() local
/aosp_15_r20/external/pytorch/c10/core/
H A DTensorImpl.cpp709 void TensorImpl::ReserveSpace(int64_t outer_dim) { in ReserveSpace()
/aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/math_ops/
H A Dsegment_reduction_ops_test.py1098 def _runGraph(self, op_functor, outer_dim, ratio, inner_dim, dtype): argument
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/reference/
H A Dreference_ops.h922 int outer_dim = std::min(batch_dim, seq_dim); in ReverseSequence() local