Searched defs:GetConvolutionFlops (Results 1 – 3 of 3) sorted by relevance
84 int64_t GpuHloCostAnalysis::GetConvolutionFlops( in GetConvolutionFlops() function in xla::gpu::GpuHloCostAnalysis
21 uint64_t GetConvolutionFlops(const BHWC& dst_shape, const OHWI& weights_shape) { in GetConvolutionFlops() function
579 int64_t HloCostAnalysis::GetConvolutionFlops( in GetConvolutionFlops() function in xla::HloCostAnalysis591 int64_t HloCostAnalysis::GetConvolutionFlops(const HloInstruction* convolution, in GetConvolutionFlops() function in xla::HloCostAnalysis