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.h144 bool IsCustomCallToDnnConvolution(const HloInstruction& hlo);
Dgpu_layout_assignment.cc204 if (IsCustomCallToDnnConvolution(*instruction)) { in AddBackendConstraints()
Dgpu_conv_algorithm_picker.cc768 CHECK(IsCustomCallToDnnConvolution(*instr)); in RunOnInstruction()
823 if (IsCustomCallToDnnConvolution(*instr)) { in RunOnComputation()
Dcudnn_pad_for_convolutions.cc130 if (IsCustomCallToDnnConvolution(*instr)) { in GetRelevantConvs()
Dir_emission_utils.cc190 bool IsCustomCallToDnnConvolution(const HloInstruction& hlo) { in IsCustomCallToDnnConvolution() function
213 IsCustomCallToDnnConvolution(hlo); in ImplementedAsLibraryCall()
Dgpu_conv_padding_legalization.cc405 if (IsCustomCallToDnnConvolution(*instr)) { in RunOnComputation()
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dmhlo_to_lhlo_with_xla.cc627 if (xla::gpu::IsCustomCallToDnnConvolution(*instr)) { in EmitCustomCallOp()