Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dshape_inference.cc3042 std::vector<int64> output_dim_bounds; in InferGatherShape() local
3043 output_dim_bounds.reserve(result_rank); in InferGatherShape()
3061 output_dim_bounds.push_back(current_bound); in InferGatherShape()
3064 return ShapeUtil::MakeShape(input_shape.element_type(), output_dim_bounds); in InferGatherShape()