Searched refs:conv_functor (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | multithreaded_conv.h | 162 EigenTensorConvFunctor<float> conv_functor; in Conv() local
163 conv_functor(device, input_data, im2col_data, batches, input_height, in Conv()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | quantized_conv_ops.cc | 221 ReferenceConvFunctor<T1, T2, T3> conv_functor; in operator ()() local
222 conv_functor(context, input_data, input_batches, input_height, in operator ()()
560 ConvFunctor<T1, T2, T3> conv_functor; in Compute() local
561 conv_functor(context, input.flat<T1>().data(), batch, input_rows, in Compute()
|
| D | conv_ops_using_gemm.cc | 548 TConvFunctor conv_functor; in Compute() local
549 conv_functor(context, input.flat<T>().data(), batch, input_rows, input_cols, in Compute()
|
| D | conv_ops_fused_image_transform.cc | 857 TConvFunctor conv_functor; in Compute() local
858 conv_functor(context, input, batch, resized_rows, resized_cols, padded_rows, in Compute()
|