Searched refs:inserted_indices (Results 1 – 2 of 2) sorted by relevance
1126 std::vector<int64> inserted_indices; in InsertedOrDeleted1SizedDimensions() local1132 [&shape_pre, &shape_post, &deleted_indices, &inserted_indices]( in InsertedOrDeleted1SizedDimensions()1149 inserted_indices.push_back(modified_output_dim); in InsertedOrDeleted1SizedDimensions()1174 return std::make_tuple(true, deleted_indices, inserted_indices); in InsertedOrDeleted1SizedDimensions()
3076 std::vector<int64> inserted_indices, deleted_indices; in HandleBroadcast() local3078 inserted_indices) = in HandleBroadcast()3082 std::reverse(inserted_indices.begin(), inserted_indices.end()); in HandleBroadcast()3083 for (auto inserted_index : inserted_indices) { in HandleBroadcast()