Searched refs:DeepCopyInstruction (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_computation_test.cc | 269 auto copy = computation->DeepCopyInstruction(constant).ValueOrDie(); in TEST_F()
286 auto tuple_copy = computation->DeepCopyInstruction(tuple).ValueOrDie(); in TEST_F()
306 EXPECT_THAT(computation->DeepCopyInstruction(constant, &indices_to_copy) in TEST_F()
314 EXPECT_EQ(computation->DeepCopyInstruction(constant, &indices_to_copy) in TEST_F()
338 computation->DeepCopyInstruction(tuple, &indices_to_copy, &copies_added) in TEST_F()
355 computation->DeepCopyInstruction(tuple, &indices_to_copy, &copies_added) in TEST_F()
373 computation->DeepCopyInstruction(tuple, &indices_to_copy, &copies_added) in TEST_F()
392 auto copy = computation->DeepCopyInstruction(token).ValueOrDie(); in TEST_F()
409 auto copy = computation->DeepCopyInstruction(tuple).ValueOrDie(); in TEST_F()
|
| D | copy_insertion.cc | 138 from->parent()->DeepCopyInstruction( in DeepCopyAndAddControlEdges()
144 to->parent()->DeepCopyInstruction(to, &indices_to_copy, &to_copy_tree)); in DeepCopyAndAddControlEdges()
280 xla_while->parent()->DeepCopyInstruction(while_init, &indices_to_copy)); in AddCopiesForWhile()
327 computation->DeepCopyInstruction(root)); in AddCopiesForConditional()
377 entry->DeepCopyInstruction( in AddCopiesForAliasedInputOutputs()
395 root->parent()->DeepCopyInstruction( in AddCopiesForAliasedInputOutputs()
1079 instruction->parent()->DeepCopyInstruction( in AddSpecialCaseCopies()
1275 computation->DeepCopyInstruction(root, &indices_to_copy)); in AddCopiesForBufferAssignment()
|
| D | hlo_computation.h | 255 StatusOr<HloInstruction*> DeepCopyInstruction(
|
| D | hlo_computation.cc | 657 StatusOr<HloInstruction*> HloComputation::DeepCopyInstruction( in DeepCopyInstruction() function in xla::HloComputation
|