Home
last modified time | relevance | path

Searched defs:dims2 (Results 1 – 7 of 7) sorted by relevance

/external/eigen/unsupported/test/
Dcxx11_tensor_contraction.cpp145 Eigen::array<DimPair, 2> dims2({{DimPair(0, 1), DimPair(1, 0)}}); in test_multidims() local
297 Eigen::array<DimPair, 2> dims2 = {{DimPair(0, 2), DimPair(2, 0)}}; in test_out_of_order_contraction() local
330 Eigen::array<DimPair, 2> dims2 = {{DimPair(4, 0), DimPair(0, 1)}}; in test_consistency() local
/external/tensorflow/tensorflow/core/kernels/data/
Dconcatenate_dataset_op.cc196 auto dims2 = ts2.dim_sizes(); in MostSpecificCompatibleShape() local
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Ddirected_interleave_dataset_op.cc264 auto dims2 = ts2.dim_sizes(); in MostSpecificCompatibleShape() local
/external/tensorflow/tensorflow/lite/kernels/
Dkernel_util.cc212 int64_t dims2 = NumDimensions(input2); in CalculateShapeForBroadcast() local
/external/tensorflow/tensorflow/c/
Dc_api_test.cc2322 const int64_t dims2[] = {2, 2}; in TEST_F() local
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Dlegacy_optimized_ops.h129 inline bool AreSameDims(const Dims<4>& dims1, const Dims<4>& dims2) { in AreSameDims()
/external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/
Doptimized_ops.h273 inline bool AreSameDims(const Dims<4>& dims1, const Dims<4>& dims2) { in AreSameDims()