Home
last modified time | relevance | path

Searched refs:result_dimensions (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dquantize.h176 std::vector<int64> result_dimensions(shape.dimensions_size());
177 std::iota(result_dimensions.begin(), result_dimensions.end(), 0);
178 std::reverse(result_dimensions.begin(), result_dimensions.end());
180 return Transpose(reshaped_result, result_dimensions);
/external/tensorflow/tensorflow/compiler/xla/
Dliteral.cc733 DimensionVector result_dimensions; in Slice() local
740 result_dimensions.push_back(dimension); in Slice()
743 ShapeUtil::MakeShapeWithLayout(shape().element_type(), result_dimensions, in Slice()