Home
last modified time | relevance | path

Searched defs:dim0 (Results 1 – 11 of 11) 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/core/kernels/
Dconcat_lib_cpu.h65 const int64 dim0 = output->dimension(0); in ConcatCPUImpl() local
117 const int64 dim0 = output->dimension(0); in ConcatCPUImpl() local
156 const int64 dim0 = output->dimension(0); in ConcatSYCLImpl() local
/external/tensorflow/tensorflow/core/tpu/
Dtpu_embedding_output_layout_utils.cc74 auto* dim0 = shape.add_dim(); in ComputeOutputTensorShapes() local
/external/deqp-deps/glslang/hlsl/
DhlslParseables.cpp90 bool IsIllegalSample(const glslang::TString& name, const char* argOrder, int dim0) in IsIllegalSample()
226 … AppendTypeName(glslang::TString& s, const char* argOrder, const char* argType, int dim0, int dim1) in AppendTypeName()
376 …id(const char* cname, char retOrder, char retType, char argOrder, char argType, int dim0, int dim1) in IsValid()
974 … for (int dim0 = dim0Min; dim0 <= dim0Max; ++dim0) { // for each dim 0... in initialize() local
/external/tensorflow/tensorflow/compiler/xla/
Dliteral_util.h485 for (int64 dim0 = 0; dim0 < dim0_size; ++dim0) { in CreateR3Projected() local
511 for (int64 dim0 = 0; dim0 < dim0_size; ++dim0) { in CreateR4Projected() local
Dliteral.h1014 int64 dim0 = 0; in PopulateR2() local
/external/tensorflow/tensorflow/core/framework/
Dtensor_shape.cc669 const int64 dim0 = dim_size(i); in MergeWith() local
706 const int64 dim0 = dim_size(i); in IsCompatibleWith() local
/external/tensorflow/tensorflow/core/ops/
Dmath_ops.cc1042 const Tensor* dim0 = c->input_tensor(3); in SparseSegmentReductionGradShapeFn() local
1084 const Tensor* dim0 = c->input_tensor(3); in SparseSegmentReductionWithNumSegmentsShapeFn() local
/external/tensorflow/tensorflow/compiler/xla/service/
Dshape_inference_test.cc412 auto dim0 = window.add_dimensions(); in TEST_F() local
457 auto dim0 = window.add_dimensions(); in TEST_F() local
503 auto dim0 = window.add_dimensions(); in TEST_F() local
547 auto dim0 = window.add_dimensions(); in TEST_F() local
Dalgebraic_simplifier_test.cc2666 const int64 dim0 = 2; in TEST_F() local
2689 const int64 dim0 = 11; in TEST_F() local
2723 const int64 dim0 = 11; in TEST_F() local
2750 const int64 dim0 = 11; in TEST_F() local
/external/tensorflow/tensorflow/core/grappler/costs/
Dop_level_cost_estimator_test.cc140 void DescribeTensor1D(int dim0, OpInfo::TensorProperties* tensor) { in DescribeTensor1D()
148 void DescribeTensor4D(int dim0, int dim1, int dim2, int dim3, in DescribeTensor4D()