Home
last modified time | relevance | path

Searched defs:dim0 (Results 1 – 25 of 72) sorted by relevance

123

/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/
H A Dtranspose_utils_test.cc364 int dim0, dim1; in TEST() local
382 int dim0, dim1; in TEST() local
400 int dim0, dim1; in TEST() local
418 int dim0, dim1; in TEST() local
435 int dim0, dim1; in TEST() local
454 int dim0, dim1; in TEST() local
473 int dim0, dim1; in TEST() local
492 int dim0, dim1; in TEST() local
/aosp_15_r20/external/pytorch/benchmarks/tensorexpr/
H A Dreduction.py134 def __init__(self, mode, device, dtype, red_dim, dim0, dim1): argument
189 def __init__(self, mode, device, dtype, dim0, dim1): argument
207 def __init__(self, mode, device, dtype, dim0, dim1): argument
238 def __init__(self, mode, device, dtype, red_dim, dim0, dim1): argument
260 def __init__(self, mode, device, dtype, dim0, dim1): argument
278 def __init__(self, mode, device, dtype, dim0, dim1): argument
/aosp_15_r20/external/pytorch/aten/src/ATen/native/mkldnn/
H A DTensorShape.cpp33 Tensor mkldnn_transpose(const Tensor& self, int64_t dim0, int64_t dim1) { in mkldnn_transpose()
37 Tensor& mkldnn_transpose_(Tensor& self, int64_t dim0, int64_t dim1) { in mkldnn_transpose_()
80 Tensor mkldnn_transpose(const Tensor& self, int64_t dim0, int64_t dim1) { in mkldnn_transpose()
94 Tensor& mkldnn_transpose_(Tensor& self, int64_t dim0, int64_t dim1) { in mkldnn_transpose_()
/aosp_15_r20/external/executorch/backends/vulkan/runtime/graph/ops/impl/
H A DTranspose.cpp29 const int64_t dim0 = graph->extract_scalar<int64_t>(extra_args[2]); in resize_transpose_view_node() local
41 const int64_t dim0, in check_transpose_view_args()
61 const int64_t dim0 = graph.extract_scalar<int64_t>(dim0_ref); in add_transpose_view_node() local
/aosp_15_r20/external/executorch/kernels/portable/cpu/util/
H A Dtranspose_util.h83 int64_t dim0, in transpose_tensors()
146 int64_t dim0, in check_transpose_copy_args()
157 SizesType dim0, in get_transpose_out_target_size()
/aosp_15_r20/external/eigen/unsupported/test/
H A Dcxx11_tensor_ifft.cpp40 static void test_2D_fft_ifft_invariant(int dim0, int dim1) { in test_2D_fft_ifft_invariant()
68 static void test_3D_fft_ifft_invariant(int dim0, int dim1, int dim2) { in test_3D_fft_ifft_invariant()
100 static void test_sub_fft_ifft_invariant(int dim0, int dim1, int dim2, int dim3) { in test_sub_fft_ifft_invariant()
/aosp_15_r20/external/angle/third_party/glslang/src/glslang/HLSL/
H A DhlslParseables.cpp78 bool IsIllegalSample(const glslang::TString& name, const char* argOrder, int dim0) in IsIllegalSample()
213 … AppendTypeName(glslang::TString& s, const char* argOrder, const char* argType, int dim0, int dim1) in AppendTypeName()
334 …ar /* retOrder */, char /* retType */, char argOrder, char /* argType */, int dim0, int /* dim1 */) in IsValid() argument
909 … for (int dim0 = dim0Min; dim0 <= dim0Max; ++dim0) { // for each dim 0... in initialize() local
/aosp_15_r20/external/deqp-deps/glslang/glslang/HLSL/
DhlslParseables.cpp78 bool IsIllegalSample(const glslang::TString& name, const char* argOrder, int dim0) in IsIllegalSample()
213 … AppendTypeName(glslang::TString& s, const char* argOrder, const char* argType, int dim0, int dim1) in AppendTypeName()
334 …ar /* retOrder */, char /* retType */, char argOrder, char /* argType */, int dim0, int /* dim1 */) in IsValid() argument
909 … for (int dim0 = dim0Min; dim0 <= dim0Max; ++dim0) { // for each dim 0... in initialize() local
/aosp_15_r20/external/pytorch/benchmarks/static_runtime/
H A Dtest_generated_ops.cc317 auto dim0 = 1; in TEST() local
1099 auto dim0 = 1; in TEST() local
1134 auto dim0 = 1; in TEST() local
1573 auto dim0 = 0; in TEST() local
1967 auto dim0 = 1; in TEST() local
2002 auto dim0 = 1; in TEST() local
2037 auto dim0 = 1; in TEST() local
2069 auto dim0 = 1; in TEST() local
2817 auto dim0 = 1; in TEST() local
2852 auto dim0 = 1; in TEST() local
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dsplit_v_op_test.cc93 #define BM_SPLITV_2D(split_dim, num_split, dim0, dim1) \ argument
110 #define BM_SPLITV_3D(split_dim, num_split, dim0, dim1, dim2) \ argument
H A Dconcat_lib_cpu.h66 const int64_t dim0 = output->dimension(0); in ConcatCPUImpl() local
118 const int64_t dim0 = output->dimension(0); in ConcatCPUImpl() local
H A Dbroadcast_to_op_test.cc27 static Graph* BroadcastTo(int dim0, int dim1, InputShape input_shape) { in BroadcastTo()
/aosp_15_r20/external/pytorch/aten/src/ATen/
H A DExpandUtils.cpp171 auto should_swap = [&](size_t dim0, size_t dim1) { in infer_dense_strides()
200 auto dim0 = i - j; in infer_dense_strides() local
H A DTensorIterator.cpp254 auto should_swap = [&](size_t dim0, size_t dim1) { in reorder_dimensions()
294 for (int dim0 = i - 1; dim0 >= 0; dim0--) { in reorder_dimensions() local
644 auto can_coalesce = [&](int dim0, int dim1) { in coalesce_dimensions()
660 auto replace_stride = [&](int dim0, int dim1) { in coalesce_dimensions()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/
H A Dliteral_util.h444 for (int64_t dim0 = 0; dim0 < dim0_size; ++dim0) { in CreateR3Projected() local
470 for (int64_t dim0 = 0; dim0 < dim0_size; ++dim0) { in CreateR4Projected() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DTensorShape.cpp898 int64_t dim0 = 1; in block_diag() local
926 int64_t dim0 = tensor.size(0); in block_diag() local
2859 static inline Tensor & sparse_transpose_(Tensor & self, int64_t dim0, int64_t dim1) { in sparse_transpose_()
2935 int64_t dim0, in propagate_transposed_names()
2945 Tensor transpose(const Tensor& self, Dimname dim0, Dimname dim1) { in transpose()
2951 Tensor & transpose_(Tensor & self, int64_t dim0, int64_t dim1) { in transpose_()
2993 int64_t dim0, in sparse_compressed_transpose()
3114 Tensor transpose(const Tensor & self, int64_t dim0, int64_t dim1) { in transpose()
3933 Tensor swapdims(const Tensor& self, int64_t dim0, int64_t dim1) { in swapdims()
3937 Tensor& swapdims_(Tensor& self, int64_t dim0, int64_t dim1) { in swapdims_()
/aosp_15_r20/external/executorch/backends/vulkan/runtime/api/containers/
H A DTensor.cpp776 const int64_t dim0, in transpose_dim_order_inplace()
787 void vTensor::virtual_transpose(const int64_t dim0, const int64_t dim1) { in virtual_transpose()
/aosp_15_r20/packages/modules/NeuralNetworks/common/types/operations/src/
DReverse.cpp44 auto dim0 = inputAxisTensorShape.dimensions[0]; in validate() local
/aosp_15_r20/external/tensorflow/tensorflow/core/ops/
H A Dmath_ops.cc1212 const Tensor* dim0 = c->input_tensor(3); in SparseSegmentReductionGradShapeFn() local
1254 const Tensor* dim0 = c->input_tensor(3); in SparseSegmentReductionWithNumSegmentsShapeFn() local
/aosp_15_r20/external/executorch/kernels/portable/cpu/
H A Dop_transpose_copy.cpp30 int64_t dim0, in transpose_copy_int_out()
/aosp_15_r20/external/tensorflow/tensorflow/core/tpu/
H A Dtpu_embedding_output_layout_utils.cc49 auto* dim0 = shape.add_dim(); in ComputeOutputTensorShapes() local
/aosp_15_r20/external/armnn/src/armnn/layers/
H A DElementwiseBaseLayer.cpp42 unsigned int dim0 = input0[i]; in InferOutputShapes() local
H A DLogicalBinaryLayer.cpp46 unsigned int dim0 = input0[i]; in InferOutputShapes() local
H A DComparisonLayer.cpp54 unsigned int dim0 = input0[i]; in InferOutputShapes() local
H A DElementwiseBinaryLayer.cpp50 unsigned int dim0 = input0[i]; in InferOutputShapes() local

123