Home
last modified time | relevance | path

Searched defs:dims1 (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/xla/tests/
Dmlir_hlo_builder_test.cc166 auto* dims1 = padding_config.add_dimensions(); in TEST_F() local
/external/tensorflow/tensorflow/core/kernels/data/
Dconcatenate_dataset_op.cc201 auto dims1 = ts1.dim_sizes(); in MostSpecificCompatibleShape() local
/external/tensorflow/tensorflow/lite/kernels/
Dkernel_util.cc409 int dims1 = NumDimensions(input1); in CalculateShapeForBroadcast() local
439 int dims1 = NumDimensions(input1); in CalculateShapeForBroadcast() local
/external/tensorflow/tensorflow/lite/delegates/flex/
Dkernel_test.cc393 TfLiteIntArray* dims1 = TfLiteIntArrayCreate(1); in TEST() local
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Ddirected_interleave_dataset_op.cc263 auto dims1 = ts1.dim_sizes(); in MostSpecificCompatibleShape() local
/external/eigen/unsupported/test/
Dcxx11_tensor_contraction.cpp329 Eigen::array<DimPair, 2> dims1 = {{DimPair(0, 4), DimPair(1, 0)}}; in test_consistency() local
/external/tensorflow/tensorflow/c/
Dc_api_test.cc2365 const int64_t dims1[] = {1, 2}; in TEST_F() local
/external/tensorflow/tensorflow/compiler/xla/service/
Dalgebraic_simplifier_test.cc609 std::vector<int64> dims1({1, 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()