Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dbfloat16_normalization_test.cc232 builder.AddInstruction(HloInstruction::CreateCrossReplicaSum( in TEST_F()
Dhlo_instruction.h287 static std::unique_ptr<HloInstruction> CreateCrossReplicaSum(
Dhlo_instruction.cc390 HloInstruction::CreateCrossReplicaSum( in CreateCrossReplicaSum() function in xla::HloInstruction
1305 clone = CreateCrossReplicaSum(shape, new_operands); in CloneWithNewOperands()
Duser_computation.cc3002 hlo_instruction = add_instruction(HloInstruction::CreateCrossReplicaSum( in Visit()
/external/tensorflow/tensorflow/compiler/xla/tools/parser/
Dhlo_parser.cc560 HloInstruction::CreateCrossReplicaSum(shape, operands)); in ParseInstruction()