Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/tests/
Dcustom_call_test.cc88 HloInstruction::CreateCustomCall(r0f32_, {constant}, "R0F32Add2")); in XLA_TEST_F()
109 HloInstruction::CreateCustomCall(r0f32_, {constant}, "R2F32ReduceSum")); in XLA_TEST_F()
124 auto incremented = b.AddInstruction(HloInstruction::CreateCustomCall( in XLA_TEST_F()
126 auto incremented_again = b.AddInstruction(HloInstruction::CreateCustomCall( in XLA_TEST_F()
148 HloInstruction::CreateCustomCall(r2f32_, {input}, "Add1ToValues")); in XLA_TEST_F()
175 auto custom_call = b.AddInstruction(HloInstruction::CreateCustomCall( in XLA_TEST_F()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcudnn_batchnorm_rewriter.cc82 std::unique_ptr<HloInstruction> libcall = HloInstruction::CreateCustomCall( in HandleBatchNormInference()
119 computation_->AddInstruction(HloInstruction::CreateCustomCall( in HandleBatchNormTraining()
200 std::unique_ptr<HloInstruction> libcall = HloInstruction::CreateCustomCall( in HandleBatchNormGrad()
Dgpu_layout_assignment_test.cc146 auto* batchnorm = builder.AddInstruction(HloInstruction::CreateCustomCall( in TEST_F()
216 auto* batchnorm = builder.AddInstruction(HloInstruction::CreateCustomCall( in TEST_F()
295 builder.AddInstruction(HloInstruction::CreateCustomCall( in TEST_F()
Dcusolver_rewriter.cc123 computation->AddInstruction(HloInstruction::CreateCustomCall( in CreateCholesky()
Dcudnn_fused_conv_rewriter.cc224 auto new_conv = computation->AddInstruction(HloInstruction::CreateCustomCall( in TryRewriteToCudnnForwardRelu()
Dcudnn_conv_rewriter.cc59 HloInstruction::CreateCustomCall(call_shape, {lhs, rhs}, call_target)); in CreateCudnnConv()
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_matchers_test.cc83 auto call = HloInstruction::CreateCustomCall( in TEST()
Dhlo_instruction_test.cc1696 auto instr1 = HloInstruction::CreateCustomCall(ShapeUtil::MakeShape(F32, {}), in TEST_F()
1708 auto instr1 = HloInstruction::CreateCustomCall(ShapeUtil::MakeShape(F32, {}), in TEST_F()
1721 auto instr = HloInstruction::CreateCustomCall(ShapeUtil::MakeShape(F32, {}), in TEST_F()
1732 auto instr = HloInstruction::CreateCustomCall(ShapeUtil::MakeShape(F32, {}), in TEST_F()
Dhlo_instruction.h765 static std::unique_ptr<HloInstruction> CreateCustomCall(
773 static std::unique_ptr<HloInstruction> CreateCustomCall(
Dhlo_instruction.cc454 CreateCustomCall(shape, all_operands(), proto.custom_call_target(), in CreateFromProto()
458 CreateCustomCall(shape, all_operands(), proto.custom_call_target(), in CreateFromProto()
1280 /* static */ std::unique_ptr<HloInstruction> HloInstruction::CreateCustomCall( in CreateCustomCall() function in xla::HloInstruction
1287 /* static */ std::unique_ptr<HloInstruction> HloInstruction::CreateCustomCall( in CreateCustomCall() function in xla::HloInstruction
Dhlo_parser.cc1539 instruction = builder->AddInstruction(HloInstruction::CreateCustomCall( in ParseInstructionRhs()
1543 instruction = builder->AddInstruction(HloInstruction::CreateCustomCall( in ParseInstructionRhs()
Dbuffer_assignment_test.cc1366 auto custom_call = builder.AddInstruction(HloInstruction::CreateCustomCall( in TEST_F()