Searched refs:inserted_indices (Results 1 – 2 of 2) sorted by relevance
997 std::vector<int64> inserted_indices; in InsertedOrDeleted1SizedDimensions() local1003 [&shape_pre, &shape_post, &deleted_indices, &inserted_indices]( in InsertedOrDeleted1SizedDimensions()1020 inserted_indices.push_back(modified_output_dim); in InsertedOrDeleted1SizedDimensions()1045 return std::make_tuple(true, deleted_indices, inserted_indices); in InsertedOrDeleted1SizedDimensions()
2065 std::vector<int64> inserted_indices, deleted_indices; in HandleBroadcast() local2067 inserted_indices) = in HandleBroadcast()2071 std::reverse(inserted_indices.begin(), inserted_indices.end()); in HandleBroadcast()2072 for (auto inserted_index : inserted_indices) { in HandleBroadcast()