Searched refs:new_scalar_indexed_source_shape (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | indexed_array_analysis.cc | 708 std::vector<int64> new_scalar_indexed_source_shape(shape.dimensions().begin(), in FoldReshapeOfGatherNoDegenerateDims() local
714 EraseAt(&new_scalar_indexed_source_shape, output_dim_after_reshape); in FoldReshapeOfGatherNoDegenerateDims()
730 /*result_shape=*/new_scalar_indexed_source_shape, in FoldReshapeOfGatherNoDegenerateDims()
760 << StrJoin(new_scalar_indexed_source_shape, ",") << "]"; in FoldReshapeOfGatherNoDegenerateDims()
765 &new_scalar_indexed_source_shape, source_dim_for_new_scalar_indexed_node, in FoldReshapeOfGatherNoDegenerateDims()
768 CHECK_EQ(absl::c_accumulate(new_scalar_indexed_source_shape, 1LL, in FoldReshapeOfGatherNoDegenerateDims()
776 /*result_shape=*/new_scalar_indexed_source_shape), in FoldReshapeOfGatherNoDegenerateDims()
791 new_scalar_indexed_source_shape))); in FoldReshapeOfGatherNoDegenerateDims()
|