Home
last modified time | relevance | path

Searched refs:OutputKernelWrapper (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dconv_ops_fused_impl.h113 OutputKernelWrapper output_kernel_wrapper( in operator()
130 functor::MatMulConvFunctor<CPUDevice, T, OutputKernelWrapper>()( in operator()
147 functor::MatMulConvFunctor<CPUDevice, T, OutputKernelWrapper>()( in operator()
156 functor::SpatialConvolution<CPUDevice, T, OutputKernelWrapper>()( in operator()
166 functor::SpatialConvolution<CPUDevice, T, OutputKernelWrapper>()( in operator()
182 struct OutputKernelWrapper { struct
188 explicit OutputKernelWrapper(OutputKernelFn fn) in OutputKernelWrapper() argument
H A Dmatmul_op_fused.cc95 OutputKernelWrapper output_kernel_wrapper( in operator ()()
147 struct OutputKernelWrapper { struct
153 explicit OutputKernelWrapper(OutputKernelFn fn) in OutputKernelWrapper() argument