| /external/pytorch/aten/src/ATen/native/mkldnn/ |
| D | TensorShape.cpp | 33 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_()
|
| /external/executorch/backends/vulkan/runtime/graph/ops/impl/ |
| D | Transpose.cpp | 29 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
|
| /external/executorch/kernels/portable/cpu/util/ |
| D | transpose_util.h | 83 int64_t dim0, in transpose_tensors()
146 int64_t dim0, in check_transpose_copy_args()
157 SizesType dim0, in get_transpose_out_target_size()
|
| /external/eigen/unsupported/test/ |
| D | cxx11_tensor_ifft.cpp | 40 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()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| D | transpose_utils_test.cc | 364 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
|
| D | transpose_utils.cc | 21 const RuntimeShape& input_shape, int* dim0, in IsTranspose2DApplicable()
|
| /external/pytorch/benchmarks/tensorexpr/ |
| D | reduction.py | 134 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
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | split_v_op_test.cc | 93 #define BM_SPLITV_2D(split_dim, num_split, dim0, dim1) \ argument
110 #define BM_SPLITV_3D(split_dim, num_split, dim0, dim1, dim2) \ argument
|
| D | concat_lib_cpu.h | 66 const int64_t dim0 = output->dimension(0); in ConcatCPUImpl() local
118 const int64_t dim0 = output->dimension(0); in ConcatCPUImpl() local
|
| D | broadcast_to_op_test.cc | 27 static Graph* BroadcastTo(int dim0, int dim1, InputShape input_shape) { in BroadcastTo()
|
| D | fingerprint_op.cc | 93 const int64_t dim0 = input.shape().dim_size(0); in Compute() local
|
| D | concat_lib_cpu.cc | 68 const auto dim0 = input->dimension(0); in EstimateBytesPerElement() local
|
| /external/tensorflow/tensorflow/core/tpu/ |
| D | tpu_embedding_output_layout_utils.cc | 49 auto* dim0 = shape.add_dim(); in ComputeOutputTensorShapes() local
|
| /external/executorch/kernels/portable/cpu/ |
| D | op_transpose_copy.cpp | 30 int64_t dim0, in transpose_copy_int_out()
|
| /external/angle/third_party/glslang/src/glslang/HLSL/ |
| D | hlslParseables.cpp | 78 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
|
| /external/deqp-deps/glslang/glslang/HLSL/ |
| D | hlslParseables.cpp | 78 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
|
| /external/pytorch/aten/src/ATen/ |
| D | ExpandUtils.cpp | 171 auto should_swap = [&](size_t dim0, size_t dim1) { in infer_dense_strides()
200 auto dim0 = i - j; in infer_dense_strides() local
|
| /external/armnn/src/armnn/layers/ |
| D | ElementwiseBaseLayer.cpp | 42 unsigned int dim0 = input0[i]; in InferOutputShapes() local
|
| D | LogicalBinaryLayer.cpp | 46 unsigned int dim0 = input0[i]; in InferOutputShapes() local
|
| D | ComparisonLayer.cpp | 54 unsigned int dim0 = input0[i]; in InferOutputShapes() local
|
| D | ElementwiseBinaryLayer.cpp | 50 unsigned int dim0 = input0[i]; in InferOutputShapes() local
|
| /external/tensorflow/tensorflow/core/tpu/ops/ |
| D | tpu_embedding_shape_util.cc | 52 auto* dim0 = shape->add_dim(); in ComputeOneTableShape() local
|
| /external/armnn/src/backends/reference/workloads/ |
| D | Slice.cpp | 51 unsigned int dim0 = paddedInput[0]; in Slice() local
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | literal_util.h | 444 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
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | data_format_ops.cc | 114 const int dim0 = input_tensor_shape.dim_size(0); in Compile() local
|