Home
last modified time | relevance | path

Searched refs:CreateCustomCall (Results 1 – 18 of 18) 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/
Dgemm_rewriter.cc61 HloInstruction::CreateCustomCall(output_shape, {lhs, rhs}, in HandleDot()
113 HloInstruction::CreateCustomCall( in HandleAdd()
Dcudnn_batchnorm_rewriter.cc105 auto batchnorm_inference_result = HloInstruction::CreateCustomCall( in HandleBatchNormInference()
164 computation_->AddInstruction(HloInstruction::CreateCustomCall( in HandleBatchNormTraining()
275 computation_->AddInstruction(HloInstruction::CreateCustomCall( in HandleBatchNormGrad()
Dgpu_layout_assignment_test.cc148 auto* batchnorm = builder.AddInstruction(HloInstruction::CreateCustomCall( in TEST_F()
218 auto* batchnorm = builder.AddInstruction(HloInstruction::CreateCustomCall( in TEST_F()
297 builder.AddInstruction(HloInstruction::CreateCustomCall( in TEST_F()
Dcusolver_rewriter.cc92 computation->AddInstruction(HloInstruction::CreateCustomCall( in CreateCholesky()
Dcudnn_fused_conv_rewriter.cc238 auto new_conv = computation->AddInstruction(HloInstruction::CreateCustomCall( in TryRewriteToCudnnForwardRelu()
Dgpu_conv_rewriter.cc60 HloInstruction::CreateCustomCall(call_shape, {lhs, rhs}, call_target)); in CreateGpuConv()
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_dce_test.cc99 HloInstruction::CreateCustomCall(ShapeUtil::MakeShape(F32, {}), in TEST_F()
117 HloInstruction::CreateCustomCall(ShapeUtil::MakeShape(F32, {}), in TEST_F()
Dhlo_instruction_test.cc1829 auto instr1 = HloInstruction::CreateCustomCall(ShapeUtil::MakeShape(F32, {}), in TEST_F()
1841 auto instr1 = HloInstruction::CreateCustomCall(ShapeUtil::MakeShape(F32, {}), in TEST_F()
1854 auto instr1 = HloInstruction::CreateCustomCall(ShapeUtil::MakeShape(F32, {}), in TEST_F()
1866 auto instr = HloInstruction::CreateCustomCall(ShapeUtil::MakeShape(F32, {}), in TEST_F()
1877 auto instr = HloInstruction::CreateCustomCall(ShapeUtil::MakeShape(F32, {}), in TEST_F()
1890 auto instr = HloInstruction::CreateCustomCall(ShapeUtil::MakeShape(F32, {}), in TEST_F()
1903 auto instr = HloInstruction::CreateCustomCall(ShapeUtil::MakeShape(F32, {}), in TEST_F()
Dtopk_rewriter.cc159 HloInstruction::CreateCustomCall(topk_shape, {input}, "TopK")); in Run()
Dhlo_instruction.h984 static std::unique_ptr<HloInstruction> CreateCustomCall(
989 static std::unique_ptr<HloInstruction> CreateCustomCall(
996 static std::unique_ptr<HloInstruction> CreateCustomCall(
1005 static std::unique_ptr<HloInstruction> CreateCustomCall(
Dhlo_matchers_test.cc96 auto call = HloInstruction::CreateCustomCall( in TEST_F()
Dhlo_instruction.cc560 CreateCustomCall(shape, all_operands(), proto.custom_call_target(), in CreateFromProto()
564 instruction = CreateCustomCall(shape, all_operands(), computations(0), in CreateFromProto()
568 instruction = CreateCustomCall( in CreateFromProto()
573 instruction = CreateCustomCall(shape, all_operands(), in CreateFromProto()
1572 /* static */ std::unique_ptr<HloInstruction> HloInstruction::CreateCustomCall( in CreateCustomCall() function in xla::HloInstruction
1579 /* static */ std::unique_ptr<HloInstruction> HloInstruction::CreateCustomCall( in CreateCustomCall() function in xla::HloInstruction
1587 /* static */ std::unique_ptr<HloInstruction> HloInstruction::CreateCustomCall( in CreateCustomCall() function in xla::HloInstruction
1596 /* static */ std::unique_ptr<HloInstruction> HloInstruction::CreateCustomCall( in CreateCustomCall() function in xla::HloInstruction
Ddynamic_padder.cc1494 comp->AddInstruction(HloInstruction::CreateCustomCall( in InsertPadToStaticOnInstruction()
1643 return comp->AddInstruction(HloInstruction::CreateCustomCall( in ConvertToDynamic()
1676 comp->AddInstruction(HloInstruction::CreateCustomCall( in ConvertToStatic()
Dhlo_parser.cc2320 instruction = builder->AddInstruction(HloInstruction::CreateCustomCall( in ParseInstructionRhs()
2326 builder->AddInstruction(HloInstruction::CreateCustomCall( in ParseInstructionRhs()
2331 builder->AddInstruction(HloInstruction::CreateCustomCall( in ParseInstructionRhs()
2336 builder->AddInstruction(HloInstruction::CreateCustomCall( in ParseInstructionRhs()
Dtuple_points_to_analysis_test.cc652 auto ccall = builder.AddInstruction(HloInstruction::CreateCustomCall( in TEST_F()
Ddynamic_dimension_inference_test.cc1228 builder.AddInstruction(HloInstruction::CreateCustomCall( in TEST_F()
Dbuffer_assignment_test.cc1538 auto custom_call = builder.AddInstruction(HloInstruction::CreateCustomCall( in TEST_F()