HomeSort by relevance Sort by last modified time
    Searched refs:DeepCopyInstruction (Results 1 - 5 of 5) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/service/gpu/
gpu_copy_insertion.cc 41 TF_ASSIGN_OR_RETURN(copy, hlo->parent()->DeepCopyInstruction(hlo));
  /external/tensorflow/tensorflow/compiler/xla/service/
hlo_computation_test.cc 256 // Test that DeepCopyInstruction properly copies an array.
262 auto copy = computation->DeepCopyInstruction(constant).ValueOrDie();
268 // Test that DeepCopyInstruction properly copies a tuple.
279 auto tuple_copy = computation->DeepCopyInstruction(tuple).ValueOrDie();
288 // Test that DeepCopyInstruction properly handles an array when the indices to
298 EXPECT_THAT(computation->DeepCopyInstruction(constant, &indices_to_copy)
306 EXPECT_EQ(computation->DeepCopyInstruction(constant, &indices_to_copy)
313 // Test that DeepCopyInstruction properly copies elements of a tuple as
330 computation->DeepCopyInstruction(tuple, &indices_to_copy, &copies_added)
346 computation->DeepCopyInstruction(tuple, &indices_to_copy, &copies_added
    [all...]
copy_insertion.cc 109 from->parent()->DeepCopyInstruction(
115 to->parent()->DeepCopyInstruction(to, &indices_to_copy, &to_copy_tree));
226 // constructed by HloInstruction::DeepCopyInstruction.
251 xla_while->parent()->DeepCopyInstruction(while_init, &indices_to_copy));
    [all...]
hlo_computation.h 245 StatusOr<HloInstruction*> DeepCopyInstruction(
hlo_computation.cc 504 StatusOr<HloInstruction*> HloComputation::DeepCopyInstruction(

Completed in 92 milliseconds