Searched refs:dim1 (Results 1 – 2 of 2) sorted by relevance
279 const auto& dim1 = result.value_tuple().dimensions_value(0); in TEST() local280 EXPECT_EQ(2, dim1.field()); in TEST()281 ASSERT_EQ(2, dim1.value_tuple().dimensions_value_size()); in TEST()283 const auto& dim11 = dim1.value_tuple().dimensions_value(0); in TEST()286 const auto& dim12 = dim1.value_tuple().dimensions_value(1); in TEST()412 const auto& dim1 = result.value_tuple().dimensions_value(0); in TEST() local413 EXPECT_EQ(DimensionsValue::ValueCase::kValueTuple, dim1.value_case()); in TEST()414 ASSERT_EQ(3, dim1.value_tuple().dimensions_value_size()); in TEST()416 const auto& dim11 = dim1.value_tuple().dimensions_value(0); in TEST()420 const auto& dim12 = dim1.value_tuple().dimensions_value(1); in TEST()[all …]
348 uint32_t dim1 = 1; in calculateBroadcastedShape() local350 dim1 = getSizeOfDimension(in1, numberOfDims1 - i); in calculateBroadcastedShape()356 if (dim1 != dim2 && dim1 != 1 && dim2 != 1) { in calculateBroadcastedShape()358 << "First tensor: dimension " << numberOfDims1 - i << " of size " << dim1 in calculateBroadcastedShape()362 out->dimensions[maxDims - i] = (dim1 == 1) ? dim2 : dim1; in calculateBroadcastedShape()