Home
last modified time | relevance | path

Searched defs:out_dims (Results 1 – 12 of 12) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Druntime_fft_impl.h70 Eigen::DSizes<Eigen::DenseIndex, FFTRank + 1> out_dims; in EigenFftR2C() local
109 Eigen::DSizes<Eigen::DenseIndex, FFTRank + 1> out_dims; in EigenFftC2R() local
/external/tensorflow/tensorflow/core/kernels/
Dmkl_transpose_op.cc132 memory::dims out_dims = TFShapeToMklDnnDims(out_tensor->shape()); in MKLTransposeND() local
Ddiag_op.cc80 const int out_dims = num_dims / 2; in Compute() local
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Ddiag_op.cc130 const int out_dims = num_dims / 2; in Compile() local
/external/tensorflow/tensorflow/compiler/xla/
Dreference_util.h263 int64 out_dims[] = {rhs.n1(), rhs.n2(), rhs.n3()}; in Concat3D() local
298 int64 out_dims[] = {rhs.n1(), rhs.n2(), rhs.n3(), rhs.n4()}; in Concat4D() local
/external/tensorflow/tensorflow/lite/kernels/
Dkernel_util.cc213 int64_t out_dims = std::max(dims1, dims2); in CalculateShapeForBroadcast() local
/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dself_adjoint_eig_test.cc105 std::vector<int64> out_dims = shape.dimensions(); in ComputeMatmulVWVt() local
/external/tensorflow/tensorflow/core/framework/
Dtensor.cc1208 gtl::InlinedVector<int64, 4> out_dims(num_out_dims, 0); in ComputeFlatInnerDims() local
1222 gtl::InlinedVector<int64, 4> out_dims(num_out_dims, 0); in ComputeFlatOuterDims() local
Dcommon_shape_fns.cc382 std::vector<DimensionHandle> out_dims(rank); in ShapeFromDimensions() local
/external/tensorflow/tensorflow/lite/toco/
Dtooling_util.cc1446 std::vector<int>* out_dims) { in MakeArrayDims()
/external/tensorflow/tensorflow/compiler/tests/
Drandomized_tests.cc2691 std::vector<int64> out_dims = RandomDims(2, 2); in TEST_F() local
2706 std::vector<int64> out_dims = RandomDims(2, 2); in TEST_F() local
/external/tensorflow/tensorflow/compiler/xla/service/
Dalgebraic_simplifier_test.cc3372 auto out_dims = in_dims; in TEST_F() local