Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dconditional_simplifier.cc97 std::vector<const Shape*> new_tuple_shapes; in TryRemoveUnusedConditionalOperands() local
98 new_tuple_shapes.reserve(tuple_indices_to_keep.size()); in TryRemoveUnusedConditionalOperands()
101 map[i] = new_tuple_shapes.size(); in TryRemoveUnusedConditionalOperands()
102 new_tuple_shapes.push_back(&param->shape().tuple_shapes(i)); in TryRemoveUnusedConditionalOperands()
104 Shape tuple_shape = ShapeUtil::MakeTupleShapeWithPtrs(new_tuple_shapes); in TryRemoveUnusedConditionalOperands()
248 std::vector<const Shape*> new_tuple_shapes; in RemoveUnusedTupleElements() local
249 new_tuple_shapes.reserve(new_tuple_shapes_size); in RemoveUnusedTupleElements()
251 new_tuple_shapes.push_back( in RemoveUnusedTupleElements()
254 const Shape new_shape = ShapeUtil::MakeTupleShapeWithPtrs(new_tuple_shapes); in RemoveUnusedTupleElements()