Searched refs:result_dimensions (Results 1 – 2 of 2) sorted by relevance
175 std::vector<int64_t> result_dimensions(shape.dimensions_size());176 std::iota(result_dimensions.begin(), result_dimensions.end(), 0);177 std::reverse(result_dimensions.begin(), result_dimensions.end());179 return Transpose(reshaped_result, result_dimensions);
1056 DimensionVector result_dimensions; in Slice() local1063 result_dimensions.push_back(dimension); in Slice()1066 ShapeUtil::MakeShapeWithLayout(shape().element_type(), result_dimensions, in Slice()