Searched refs:MatMulConvFunctor (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | multithreaded_conv.h | 58 struct MatMulConvFunctor { struct
106 MatMulConvFunctor<Eigen::ThreadPoolDevice, T>()(device, output, input, in operator()
119 MatMulConvFunctor<Eigen::ThreadPoolDevice, T>()(device, output, input, in operator()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | conv_ops.cc | 91 functor::MatMulConvFunctor<Device, T>()( in operator ()()
107 functor::MatMulConvFunctor<Device, T>()( in operator ()()
931 void MatMulConvFunctor<GPUDevice, T>::operator()( \
937 extern template struct MatMulConvFunctor<GPUDevice, T>; \
|
| D | conv_ops_fused_impl.h | 269 functor::MatMulConvFunctor<CPUDevice, T, OutputKernel>()( in operator()
286 functor::MatMulConvFunctor<CPUDevice, T, OutputKernel>()( in operator()
|
| D | conv_2d.h | 134 struct MatMulConvFunctor {
|