Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/
Dshape_util.cc1014 for (int64 modified_output_dim = prior_unmodified_dim_pair.second + 1; in InsertedOrDeleted1SizedDimensions() local
1015 modified_output_dim < unmodified_dim_pair.second; in InsertedOrDeleted1SizedDimensions()
1016 ++modified_output_dim) { in InsertedOrDeleted1SizedDimensions()
1017 if (shape_post.dimensions(modified_output_dim) > 1) { in InsertedOrDeleted1SizedDimensions()
1020 inserted_indices.push_back(modified_output_dim); in InsertedOrDeleted1SizedDimensions()