Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/
Dshape_util.cc1003 std::vector<int64> inserted_indices; in InsertedOrDeleted1SizedDimensions() local
1009 [&shape_pre, &shape_post, &deleted_indices, &inserted_indices]( in InsertedOrDeleted1SizedDimensions()
1026 inserted_indices.push_back(modified_output_dim); in InsertedOrDeleted1SizedDimensions()
1051 return std::make_tuple(true, deleted_indices, inserted_indices); in InsertedOrDeleted1SizedDimensions()
/external/tensorflow/tensorflow/compiler/xla/service/
Dalgebraic_simplifier.cc1148 std::vector<int64> inserted_indices, deleted_indices; in HandleBroadcast() local
1150 inserted_indices) = in HandleBroadcast()
1154 std::reverse(inserted_indices.begin(), inserted_indices.end()); in HandleBroadcast()
1156 for (auto inserted_index : inserted_indices) { in HandleBroadcast()