Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dasync_collective_creator.cc72 start->CopyBackendConfigFrom(ar); in Run()
102 start->CopyBackendConfigFrom(ag); in Run()
141 collective_permute_start->CopyBackendConfigFrom(cp); in Run()
Dwhile_loop_simplifier.cc176 new_while_op->CopyBackendConfigFrom(while_op); in RemoveDeadTupleIndices()
759 new_while_op->CopyBackendConfigFrom(while_op); in TryRemoveConstantParams()
1090 new_while_op->CopyBackendConfigFrom(while_op); in TryFlattenNestedTuples()
1329 new_while->CopyBackendConfigFrom(while_op); in TryMergeInductionVariables()
Dhlo_computation.cc788 async_start->CopyBackendConfigFrom(instruction); in CreateAsyncInstructions()
790 async_done->CopyBackendConfigFrom(instruction); in CreateAsyncInstructions()
Dhlo_instruction.h1747 void CopyBackendConfigFrom(const HloInstruction* other) { in CopyBackendConfigFrom() function
/external/tensorflow/tensorflow/compiler/xla/tools/
Dhlo_control_flow_flattening.cc344 custom_call->CopyBackendConfigFrom(hlo); in RemoveCollective()