Searched refs:CreateCustomCall (Results 1 – 12 of 12) sorted by relevance
/external/tensorflow/tensorflow/compiler/xla/tests/ |
D | custom_call_test.cc | 88 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/ |
D | cudnn_batchnorm_rewriter.cc | 82 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()
|
D | gpu_layout_assignment_test.cc | 146 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()
|
D | cusolver_rewriter.cc | 123 computation->AddInstruction(HloInstruction::CreateCustomCall( in CreateCholesky()
|
D | cudnn_fused_conv_rewriter.cc | 224 auto new_conv = computation->AddInstruction(HloInstruction::CreateCustomCall( in TryRewriteToCudnnForwardRelu()
|
D | cudnn_conv_rewriter.cc | 59 HloInstruction::CreateCustomCall(call_shape, {lhs, rhs}, call_target)); in CreateCudnnConv()
|
/external/tensorflow/tensorflow/compiler/xla/service/ |
D | hlo_matchers_test.cc | 83 auto call = HloInstruction::CreateCustomCall( in TEST()
|
D | hlo_instruction_test.cc | 1696 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()
|
D | hlo_instruction.h | 765 static std::unique_ptr<HloInstruction> CreateCustomCall( 773 static std::unique_ptr<HloInstruction> CreateCustomCall(
|
D | hlo_instruction.cc | 454 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
|
D | hlo_parser.cc | 1539 instruction = builder->AddInstruction(HloInstruction::CreateCustomCall( in ParseInstructionRhs() 1543 instruction = builder->AddInstruction(HloInstruction::CreateCustomCall( in ParseInstructionRhs()
|
D | buffer_assignment_test.cc | 1366 auto custom_call = builder.AddInstruction(HloInstruction::CreateCustomCall( in TEST_F()
|