Searched refs:new_scalar_indexed_source_shape (Results 1 – 1 of 1) sorted by relevance
714 std::vector<int64_t> new_scalar_indexed_source_shape( in FoldReshapeOfGatherNoDegenerateDims() local720 EraseAt(&new_scalar_indexed_source_shape, output_dim_after_reshape); in FoldReshapeOfGatherNoDegenerateDims()735 /*result_shape=*/new_scalar_indexed_source_shape, in FoldReshapeOfGatherNoDegenerateDims()765 << StrJoin(new_scalar_indexed_source_shape, ",") << "]"; in FoldReshapeOfGatherNoDegenerateDims()770 &new_scalar_indexed_source_shape, source_dim_for_new_scalar_indexed_node, in FoldReshapeOfGatherNoDegenerateDims()773 CHECK_EQ(absl::c_accumulate(new_scalar_indexed_source_shape, 1LL, in FoldReshapeOfGatherNoDegenerateDims()780 /*result_shape=*/new_scalar_indexed_source_shape), in FoldReshapeOfGatherNoDegenerateDims()795 new_scalar_indexed_source_shape))); in FoldReshapeOfGatherNoDegenerateDims()