Home
last modified time | relevance | path

Searched defs:GetConvolutionFlops (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dgpu_hlo_cost_analysis.cc84 int64_t GpuHloCostAnalysis::GetConvolutionFlops( in GetConvolutionFlops() function in xla::gpu::GpuHloCostAnalysis
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/
H A Dflops_util.cc21 uint64_t GetConvolutionFlops(const BHWC& dst_shape, const OHWI& weights_shape) { in GetConvolutionFlops() function
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dhlo_cost_analysis.cc579 int64_t HloCostAnalysis::GetConvolutionFlops( in GetConvolutionFlops() function in xla::HloCostAnalysis
591 int64_t HloCostAnalysis::GetConvolutionFlops(const HloInstruction* convolution, in GetConvolutionFlops() function in xla::HloCostAnalysis