Searched refs:custom_call_target (Results 1 – 14 of 14) sorted by relevance
/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
D | gpu_layout_assignment.cc | 51 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()
|
D | cudnn_batchnorm_thunk.cc | 95 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()
|
D | pad_insertion.cc | 30 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()
|
D | ir_emission_utils.cc | 104 const auto& target = hlo.custom_call_target(); in IsCustomCallToDnnBatchNorm() 120 const auto& target = hlo.custom_call_target(); in IsCustomCallToDnnConvolution()
|
D | cudnn_convolution_algorithm_picker.cc | 267 const auto& call_target = instr->custom_call_target(); in RunOnInstruction() 324 instr->custom_call_target())); in RunOnInstruction()
|
D | ir_emitter_unnested.cc | 324 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/ |
D | hlo_matchers.cc | 119 call_target_matcher_, instruction->custom_call_target(), &sub_listener); in MatchAndExplain()
|
D | hlo_tfgraph_builder.cc | 198 attrs["custom_call_target"].set_s(instruction->custom_call_target()); in SetNodeAttrs()
|
D | hlo_instruction.h | 482 tensorflow::StringPiece custom_call_target); 743 const string& custom_call_target() const;
|
D | hlo.proto | 119 string custom_call_target = 28; field
|
D | hlo_instruction.cc | 146 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/ |
D | hlo_parser.cc | 987 optional<string> custom_call_target; in ParseInstruction() local 989 &custom_call_target}; in ParseInstruction() 994 shape, operands, *custom_call_target)); in ParseInstruction()
|
D | hlo_parser_test.cc | 836 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/ |
D | ir_emitter.cc | 2189 tensorflow::StringPiece custom_call_target(custom_call->custom_call_target()); in HandleCustomCall() local 2205 AsStringRef(custom_call_target), in HandleCustomCall()
|