Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/lite/kernels/internal/
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
Dtranspose_utils.cc21 const RuntimeShape& input_shape, int* dim0, in IsTranspose2DApplicable()
/external/eigen/unsupported/test/
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()
/external/tensorflow/tensorflow/core/kernels/
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
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
Dbroadcast_to_op_test.cc27 static Graph* BroadcastTo(int dim0, int dim1, InputShape input_shape) { in BroadcastTo()
Dfingerprint_op.cc93 const int64_t dim0 = input.shape().dim_size(0); in Compute() local
Dconcat_lib_cpu.cc68 const auto dim0 = input->dimension(0); in EstimateBytesPerElement() local
/external/tensorflow/tensorflow/core/tpu/
Dtpu_embedding_output_layout_utils.cc49 auto* dim0 = shape.add_dim(); in ComputeOutputTensorShapes() local
/external/angle/third_party/vulkan-deps/glslang/src/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
/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
/external/armnn/src/armnn/layers/
DElementwiseBaseLayer.cpp42 unsigned int dim0 = input0[i]; in InferOutputShapes() local
DLogicalBinaryLayer.cpp46 unsigned int dim0 = input0[i]; in InferOutputShapes() local
DComparisonLayer.cpp54 unsigned int dim0 = input0[i]; in InferOutputShapes() local
DElementwiseBinaryLayer.cpp50 unsigned int dim0 = input0[i]; in InferOutputShapes() local
/external/tensorflow/tensorflow/core/tpu/ops/
Dtpu_embedding_shape_util.cc52 auto* dim0 = shape->add_dim(); in ComputeOneTableShape() local
/external/tensorflow/tensorflow/compiler/xla/
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
/external/armnn/src/backends/reference/workloads/
DSlice.cpp51 unsigned int dim0 = paddedInput[0]; in Slice() local
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Ddata_format_ops.cc114 const int dim0 = input_tensor_shape.dim_size(0); in Compile() local
/external/tensorflow/tensorflow/core/ops/
Dmath_ops.cc1212 const Tensor* dim0 = c->input_tensor(3); in SparseSegmentReductionGradShapeFn() local
1254 const Tensor* dim0 = c->input_tensor(3); in SparseSegmentReductionWithNumSegmentsShapeFn() local
/external/ComputeLibrary/arm_compute/core/
DUtils.h1222 inline unsigned int adjust_vec_size(unsigned int vec_size, size_t dim0) in adjust_vec_size()
/external/tensorflow/tensorflow/core/framework/
Dtensor_shape.cc923 const int64_t dim0 = dim_size(i); in MergeWith() local
963 const int64_t dim0 = dim_size(i); in IsCompatibleWith() local
/external/tensorflow/tensorflow/python/keras/engine/
Dkeras_tensor.py538 def __init__(self, tensor, dim0): argument
/external/tensorflow/tensorflow/compiler/xla/service/
Dshape_inference_test.cc450 auto dim0 = window.add_dimensions(); in TEST_F() local
495 auto dim0 = window.add_dimensions(); in TEST_F() local
541 auto dim0 = window.add_dimensions(); in TEST_F() local
585 auto dim0 = window.add_dimensions(); in TEST_F() local
620 auto dim0 = window.add_dimensions(); in TEST_F() local
669 auto dim0 = args.window.add_dimensions(); in MakeConvolveArgs() local
/external/tensorflow/tensorflow/core/grappler/costs/
Dop_level_cost_estimator_test.cc169 void DescribeTensor1D(int dim0, OpInfo::TensorProperties* tensor) { in DescribeTensor1D()
177 void DescribeTensor4D(int dim0, int dim1, int dim2, int dim3, in DescribeTensor4D()
189 void DescribeTensor5D(int dim0, int dim1, int dim2, int dim3, int dim4, in DescribeTensor5D()

12