Home
last modified time | relevance | path

Searched refs:IsCustomCallToDnnConvolution (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emission_utils.h132 bool IsCustomCallToDnnConvolution(const HloInstruction& hlo);
Dir_emission_utils.cc134 bool IsCustomCallToDnnConvolution(const HloInstruction& hlo) { in IsCustomCallToDnnConvolution() function
157 IsCustomCallToDnnConvolution(hlo); in ImplementedAsLibraryCall()
Dcudnn_conv_algorithm_picker.cc360 CHECK(IsCustomCallToDnnConvolution(*instr)); in RunOnInstruction()
413 if (IsCustomCallToDnnConvolution(*instr)) { in RunOnComputation()
Dcudnn_conv_pad_for_tensor_cores.cc224 if (IsCustomCallToDnnConvolution(*instr)) { in GetRelevantConvs()
Dgpu_layout_assignment.cc182 if (IsCustomCallToDnnConvolution(*instruction)) { in AddBackendConstraints()
Dcudnn_conv_padding_legalization.cc397 if (IsCustomCallToDnnConvolution(*instr)) { in RunOnComputation()
Dir_emitter_unnested.cc467 if (IsCustomCallToDnnConvolution(*custom_call)) { in HandleCustomCall()