| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_layout_assignment.cc | 84 instr->custom_call_target() == kCudnnConvBackwardInputCallTarget && in HeuristicLayoutAssignment()
282 instruction->custom_call_target() == in PropagateOperandConstraint()
294 instruction->custom_call_target() == in PropagateOperandConstraint()
309 instruction->custom_call_target() == kCudnnBatchNormBackwardCallTarget && in PropagateOperandConstraint()
340 instruction->custom_call_target() == in PropagateBufferConstraint()
347 instruction->custom_call_target() == in PropagateBufferConstraint()
354 instruction->custom_call_target() == kCudnnBatchNormBackwardCallTarget && in PropagateBufferConstraint()
|
| D | ir_emission_utils.cc | 120 const auto& target = hlo.custom_call_target(); in IsCustomCallToDnnBatchNorm()
138 const auto& target = hlo.custom_call_target(); in IsCustomCallToDnnConvolution()
151 const auto& target = hlo.custom_call_target(); in IsCustomCallToCusolver()
249 absl::string_view target = instr->custom_call_target(); in GetCudnnConvKind()
|
| D | cudnn_batchnorm_thunk.cc | 94 CHECK_EQ(hlo->custom_call_target(), in CudnnBatchNormForwardInferenceThunk()
154 CHECK_EQ(hlo->custom_call_target(), kCudnnBatchNormForwardTrainingCallTarget); in CudnnBatchNormForwardTrainingThunk()
236 CHECK_EQ(hlo->custom_call_target(), kCudnnBatchNormBackwardCallTarget); in CudnnBatchNormBackwardThunk()
|
| D | cudnn_conv_padding_legalization.cc | 34 CHECK(conv.custom_call_target() == kCudnnConvForwardCallTarget || in IsForwardConvolutionCanonical()
35 conv.custom_call_target() == kCudnnConvBiasActivationForwardCallTarget); in IsForwardConvolutionCanonical()
204 CHECK_EQ(backward_conv->custom_call_target(), in CanonicalizeBackwardFilterConvolution()
|
| D | cudnn_fused_conv_rewriter.cc | 148 if (conv->custom_call_target() != kCudnnConvForwardCallTarget) { in FindConvWithRelu()
257 if (call->custom_call_target() == kCudnnConvForwardCallTarget) { in Run()
|
| D | ir_emitter_unnested.cc | 380 if (custom_call->custom_call_target() == in HandleCustomCall()
404 if (custom_call->custom_call_target() == in HandleCustomCall()
435 if (custom_call->custom_call_target() == kCudnnBatchNormBackwardCallTarget) { in HandleCustomCall()
484 if (custom_call->custom_call_target() == kCusolverCholeskyCallTarget) { in HandleCustomCall()
|
| /external/tensorflow/tensorflow/compiler/xla/service/interpreter/ |
| D | compiler.cc | 56 void* target_fn = registry->Lookup(custom_call->custom_call_target()); in HandleEvaluatorCustomCall()
59 custom_call->custom_call_target()); in HandleEvaluatorCustomCall()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_matchers.cc | 136 call_target_matcher_, instruction->custom_call_target(), &sub_listener); in MatchAndExplain()
|
| D | hlo_parser_test.cc | 906 …]{0,2,1} custom-call(f32[42,2,3]{0,1,2} %p0, f32[123,4]{0,1} %p1), custom_call_target="baz", opera… in CreateTestCases()
917 …ROOT %custom-call = f32[1,2,3]{0,2,1} custom-call(), custom_call_target="baz", operand_layout_cons… in CreateTestCases()
930 …all((f32[2,2]{0,1}, f32[42,2,3]{0,1,2}) %p0, f32[123,4]{0,1} %p1), custom_call_target="baz", opera… in CreateTestCases()
1285 ROOT custom-call = f32[1,2,3]{0,2,1} custom-call(constant), custom_call_target="foo\"bar" in CreateTestCases()
1297 …ROOT custom-call = f32[1,2,3]{0,2,1} custom-call(constant), custom_call_target="foo\"bar", opaque=… in CreateTestCases()
1457 …, window={size=2x2}, dim_labels=b01f_01io->b01f, feature_group_count=2, custom_call_target="target" in CreateTestCases()
|
| D | hlo_instructions.cc | 2046 absl::string_view custom_call_target, absl::string_view opaque) in HloCustomCallInstruction() argument
2048 custom_call_target_(custom_call_target.begin(), custom_call_target.end()), in HloCustomCallInstruction()
2060 absl::string_view custom_call_target, absl::string_view opaque, in HloCustomCallInstruction() argument
2063 custom_call_target_(custom_call_target.begin(), custom_call_target.end()), in HloCustomCallInstruction()
2180 shape, new_operands, custom_call_target(), opaque()); in CloneWithNewOperandsImpl()
|
| D | hlo_instruction.h | 767 absl::string_view custom_call_target, absl::string_view opaque = "");
775 absl::string_view custom_call_target,
1581 const string& custom_call_target() const;
|
| 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()
1282 absl::string_view custom_call_target, absl::string_view opaque) { in CreateCustomCall() argument
1284 shape, operands, custom_call_target, opaque); in CreateCustomCall()
1289 absl::string_view custom_call_target, in CreateCustomCall() argument
1293 shape, operands, custom_call_target, opaque, operand_shapes_with_layout); in CreateCustomCall()
3500 const string& HloInstruction::custom_call_target() const { in custom_call_target() function in xla::HloInstruction
3501 return Cast<HloCustomCallInstruction>(this)->custom_call_target(); in custom_call_target()
|
| D | hlo_instructions.h | 1171 absl::string_view custom_call_target,
1178 absl::string_view custom_call_target,
1202 const string& custom_call_target() const { return custom_call_target_; } in custom_call_target() function
|
| D | hlo.proto | 132 string custom_call_target = 28; field
|
| D | hlo_parser.cc | 1484 optional<string> custom_call_target; in ParseInstructionRhs() local
1492 &custom_call_target}; in ParseInstructionRhs()
1540 shape, operands, *custom_call_target, *operand_layout_constraints, in ParseInstructionRhs()
1544 shape, operands, *custom_call_target, in ParseInstructionRhs()
|
| D | hlo_evaluator_test.cc | 3238 EXPECT_EQ("_my_custom_call", custom_call->custom_call_target()); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | ir_emitter.cc | 2265 custom_call->custom_call_target(), in HandleCustomCall()
|