Home
last modified time | relevance | path

Searched refs:custom_call_target (Results 1 – 14 of 14) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_layout_assignment.cc51 const auto& target = instr->custom_call_target(); in AddBackendConstraintsToDnnConvCustomCall()
66 << instr->custom_call_target(); in AddBackendConstraintsToDnnConvCustomCall()
128 << instr->custom_call_target(); in AddBackendConstraintsToDnnConvCustomCall()
162 instruction->custom_call_target() == in PropagateOperandConstraint()
174 instruction->custom_call_target() == in PropagateOperandConstraint()
189 instruction->custom_call_target() == kCudnnBatchNormBackwardCallTarget && in PropagateOperandConstraint()
220 instruction->custom_call_target() == in PropagateBufferConstraint()
227 instruction->custom_call_target() == in PropagateBufferConstraint()
234 instruction->custom_call_target() == kCudnnBatchNormBackwardCallTarget && in PropagateBufferConstraint()
Dcudnn_batchnorm_thunk.cc95 CHECK_EQ(hlo->custom_call_target(), in CudnnBatchNormForwardInferenceThunk()
152 CHECK_EQ(hlo->custom_call_target(), kCudnnBatchNormForwardTrainingCallTarget); in CudnnBatchNormForwardTrainingThunk()
232 CHECK_EQ(hlo->custom_call_target(), kCudnnBatchNormBackwardCallTarget); in CudnnBatchNormBackwardThunk()
Dpad_insertion.cc30 CHECK_EQ(conv.custom_call_target(), kCudnnConvForwardCallTarget); in IsForwardConvolutionCanonical()
202 CHECK_EQ(backward_conv->custom_call_target(), in CanonicalizeBackwardFilterConvolution()
396 const auto& target = instruction->custom_call_target(); in Run()
Dir_emission_utils.cc104 const auto& target = hlo.custom_call_target(); in IsCustomCallToDnnBatchNorm()
120 const auto& target = hlo.custom_call_target(); in IsCustomCallToDnnConvolution()
Dcudnn_convolution_algorithm_picker.cc267 const auto& call_target = instr->custom_call_target(); in RunOnInstruction()
324 instr->custom_call_target())); in RunOnInstruction()
Dir_emitter_unnested.cc324 if (custom_call->custom_call_target() == in HandleCustomCall()
348 if (custom_call->custom_call_target() == in HandleCustomCall()
379 if (custom_call->custom_call_target() == kCudnnBatchNormBackwardCallTarget) { in HandleCustomCall()
431 const auto& target = custom_call->custom_call_target(); in HandleCustomCall()
474 << custom_call->custom_call_target(); in HandleCustomCall()
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_matchers.cc119 call_target_matcher_, instruction->custom_call_target(), &sub_listener); in MatchAndExplain()
Dhlo_tfgraph_builder.cc198 attrs["custom_call_target"].set_s(instruction->custom_call_target()); in SetNodeAttrs()
Dhlo_instruction.h482 tensorflow::StringPiece custom_call_target);
743 const string& custom_call_target() const;
Dhlo.proto119 string custom_call_target = 28; field
Dhlo_instruction.cc146 instruction->custom_call_target_ = proto.custom_call_target(); in CreateFromProto()
1125 tensorflow::StringPiece custom_call_target) { in CreateCustomCall() argument
1131 instruction->custom_call_target_ = custom_call_target.ToString(); in CreateCustomCall()
1997 const string& HloInstruction::custom_call_target() const { in custom_call_target() function in xla::HloInstruction
/external/tensorflow/tensorflow/compiler/xla/tools/parser/
Dhlo_parser.cc987 optional<string> custom_call_target; in ParseInstruction() local
989 &custom_call_target}; in ParseInstruction()
994 shape, operands, *custom_call_target)); in ParseInstruction()
Dhlo_parser_test.cc836 ROOT custom-call = f32[1,2,3]{0,2,1} custom-call(constant), custom_call_target="foo\"bar"
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.cc2189 tensorflow::StringPiece custom_call_target(custom_call->custom_call_target()); in HandleCustomCall() local
2205 AsStringRef(custom_call_target), in HandleCustomCall()