Searched refs:dim1 (Results 1 – 2 of 2) sorted by relevance
273 const auto& dim1 = result.value_tuple().dimensions_value(0); in TEST() local274 EXPECT_EQ(2, dim1.field()); in TEST()275 ASSERT_EQ(2, dim1.value_tuple().dimensions_value_size()); in TEST()277 const auto& dim11 = dim1.value_tuple().dimensions_value(0); in TEST()280 const auto& dim12 = dim1.value_tuple().dimensions_value(1); in TEST()406 const auto& dim1 = result.value_tuple().dimensions_value(0); in TEST() local407 EXPECT_EQ(DimensionsValue::ValueCase::kValueTuple, dim1.value_case()); in TEST()408 ASSERT_EQ(3, dim1.value_tuple().dimensions_value_size()); in TEST()410 const auto& dim11 = dim1.value_tuple().dimensions_value(0); in TEST()414 const auto& dim12 = dim1.value_tuple().dimensions_value(1); in TEST()[all …]
346 uint32_t dim1 = 1; in calculateBroadcastedShape() local348 dim1 = getSizeOfDimension(in1, numberOfDims1 - i); in calculateBroadcastedShape()354 if (dim1 != dim2 && dim1 != 1 && dim2 != 1) { in calculateBroadcastedShape()356 << "First tensor: dimension " << numberOfDims1 - i << " of size " << dim1 in calculateBroadcastedShape()360 out->dimensions[maxDims - i] = (dim1 == 1) ? dim2 : dim1; in calculateBroadcastedShape()