Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dshape_inference.cc3332 std::vector<int64> expanded_start_indices_shape; in InferGatherShape() local
3335 expanded_start_indices_shape.reserve(start_indices_shape.dimensions_size()); in InferGatherShape()
3339 std::back_inserter(expanded_start_indices_shape)); in InferGatherShape()
3343 if (expanded_start_indices_shape.size() == in InferGatherShape()
3345 expanded_start_indices_shape.push_back(1); in InferGatherShape()
3350 input_shape, expanded_start_indices_shape, gather_dim_numbers)); in InferGatherShape()
3392 (expanded_start_indices_shape.size() - 1); in InferGatherShape()
3432 current_bound = expanded_start_indices_shape[gather_dims_seen++]; in InferGatherShape()