Home
last modified time | relevance | path

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

/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/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/tensorflow/tensorflow/core/kernels/
Dconcat_lib_cpu.h66 const int64 dim0 = output->dimension(0); in ConcatCPUImpl() local
118 const int64 dim0 = output->dimension(0); in ConcatCPUImpl() local
Dbroadcast_to_op_test.cc27 static Graph* BroadcastTo(int dim0, int dim1, InputShape input_shape) { in BroadcastTo()
Dconcat_lib_cpu.cc68 const auto dim0 = input->dimension(0); in EstimateBytesPerElement() local
Dfingerprint_op.cc93 const int64 dim0 = input.shape().dim_size(0); in Compute() local
/external/tensorflow/tensorflow/core/tpu/
Dtpu_embedding_output_layout_utils.cc30 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/tensorflow/tensorflow/compiler/xla/
Dliteral_util.h452 for (int64 dim0 = 0; dim0 < dim0_size; ++dim0) { in CreateR3Projected() local
478 for (int64 dim0 = 0; dim0 < dim0_size; ++dim0) { in CreateR4Projected() local
Dliteral.h1016 int64 dim0 = 0; in PopulateR2() 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.cc1179 const Tensor* dim0 = c->input_tensor(3); in SparseSegmentReductionGradShapeFn() local
1221 const Tensor* dim0 = c->input_tensor(3); in SparseSegmentReductionWithNumSegmentsShapeFn() local
/external/tensorflow/tensorflow/core/framework/
Dtensor_shape.cc911 const int64 dim0 = dim_size(i); in MergeWith() local
951 const int64 dim0 = dim_size(i); in IsCompatibleWith() local
/external/tensorflow/tensorflow/python/keras/engine/
Dkeras_tensor.py557 def __init__(self, tensor, dim0): argument
/external/tensorflow/tensorflow/compiler/xla/service/
Dshape_inference_test.cc424 auto dim0 = window.add_dimensions(); in TEST_F() local
469 auto dim0 = window.add_dimensions(); in TEST_F() local
515 auto dim0 = window.add_dimensions(); in TEST_F() local
559 auto dim0 = window.add_dimensions(); in TEST_F() local
594 auto dim0 = window.add_dimensions(); in TEST_F() local
643 auto dim0 = args.window.add_dimensions(); in MakeConvolveArgs() local
Dalgebraic_simplifier_test.cc3426 const int64 dim0 = 2; in TEST_F() local
3449 const int64 dim0 = 11; in TEST_F() local
3483 const int64 dim0 = 11; in TEST_F() local
3510 const int64 dim0 = 11; in TEST_F() local
Dhlo_evaluator_test.cc178 int dim0 = expected.shape().dimensions(0); in ReduceWindowMaxIotaTest() local
/external/tensorflow/tensorflow/core/grappler/costs/
Dop_level_cost_estimator_test.cc168 void DescribeTensor1D(int dim0, OpInfo::TensorProperties* tensor) { in DescribeTensor1D()
176 void DescribeTensor4D(int dim0, int dim1, int dim2, int dim3, in DescribeTensor4D()
188 void DescribeTensor5D(int dim0, int dim1, int dim2, int dim3, int dim4, in DescribeTensor5D()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_ops_a_m.cc1906 int64_t dim0 = input_ty.getDimSize(0); in Verify() local
1912 int64_t dim0 = input_ty.getDimSize(0); in Verify() local
/external/tensorflow/tensorflow/python/framework/
Dops.py6985 def __init__(self, tensor, dim0): argument
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Doptimized_ops.h7829 int dim0, dim1; in TransposeImpl() local