Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dgpu_hlo_cost_analysis.cc55 if (IsCustomCallToDnnConvolution(*custom_call)) { in HandleCustomCall()
93 if (IsCustomCallToDnnConvolution(*convolution) && in GetConvolutionFlops()
H A Dcublas_cudnn.h108 bool IsCustomCallToDnnConvolution(const HloInstruction& hlo);
H A Dcublas_cudnn.cc42 bool IsCustomCallToDnnConvolution(const HloInstruction& hlo) { in IsCustomCallToDnnConvolution() function
H A Dgpu_conv_algorithm_picker.cc907 CHECK(IsCustomCallToDnnConvolution(*instr)); in RunOnInstruction()
986 if (IsCustomCallToDnnConvolution(*instr)) { in RunOnComputation()
H A Dgpu_conv_padding_legalization.cc409 if (IsCustomCallToDnnConvolution(*instr)) { in RunOnComputation()
H A Dgpu_layout_assignment.cc253 if (IsCustomCallToDnnConvolution(*instruction)) { in AddBackendConstraints()
H A Dcudnn_pad_for_convolutions.cc132 if (IsCustomCallToDnnConvolution(*instr)) { in GetRelevantConvs()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dhlo_graph_dumper.cc1270 if (gpu::IsCustomCallToDnnConvolution(*instr)) { in GetInstructionNodeBackendConfig()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
H A Dmhlo_to_lhlo_with_xla.cc710 if (xla::gpu::IsCustomCallToDnnConvolution(*instr)) { in EmitCustomCallOp()