Home
last modified time | relevance | path

Searched defs:kernel_filters (Results 1 – 9 of 9) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/
H A Druntime_conv2d_acl.cc33 int64_t kernel_channels, int64_t kernel_filters, int64_t output_rows, in ACLDepthwiseConvImpl()
131 int64_t kernel_channels, int64_t kernel_filters, in ACLGemmConvImpl()
237 int64_t kernel_channels, int64_t kernel_filters, int64_t output_rows, in __xla_cpu_runtime_ACLConv2DF32()
H A Druntime_single_threaded_conv2d.cc26 int64_t kernel_cols, int64_t kernel_channels, int64_t kernel_filters, in __xla_cpu_runtime_EigenSingleThreadedConv2DF16()
46 int64_t kernel_channels, int64_t kernel_filters, int64_t output_rows, in __xla_cpu_runtime_EigenSingleThreadedConv2DF32()
H A Druntime_conv2d.cc29 int64_t kernel_channels, int64_t kernel_filters, int64_t output_rows, in __xla_cpu_runtime_EigenConv2DF32()
51 int64_t kernel_cols, int64_t kernel_channels, int64_t kernel_filters, in __xla_cpu_runtime_EigenConv2DF16()
H A Druntime_conv3d.cc29 int64_t kernel_z, int64_t kernel_channels, int64_t kernel_filters, in __xla_cpu_runtime_EigenConv3DF32()
54 int64_t kernel_z, int64_t kernel_channels, int64_t kernel_filters, in __xla_cpu_runtime_EigenConv3DF16()
H A Druntime_single_threaded_conv3d.cc26 int64_t kernel_z, int64_t kernel_channels, int64_t kernel_filters, in __xla_cpu_runtime_EigenSingleThreadedConv3DF32()
49 int64_t kernel_z, int64_t kernel_channels, int64_t kernel_filters, in __xla_cpu_runtime_EigenSingleThreadedConv3DF16()
H A Druntime_conv2d_mkl.cc56 int64_t kernel_channels, int64_t kernel_filters, in MKLConvImpl()
158 int64_t kernel_channels, int64_t kernel_filters, int64_t output_rows, in __xla_cpu_runtime_MKLConv2DF32()
H A Druntime_conv_impl.h35 Eigen::Index kernel_filters, Eigen::Index output_x, Eigen::Index output_y, in EigenConv2DImpl()
112 Eigen::Index kernel_channels, Eigen::Index kernel_filters, in EigenConv3DImpl()
H A Druntime_conv2d_acl.h81 int64_t kernel_cols, int64_t kernel_channels, int64_t kernel_filters, in __xla_cpu_runtime_ACLConv2DF32()
H A Dir_emitter.cc962 int64_t kernel_filters = in HandleConvolution() local