Home
last modified time | relevance | path

Searched defs:kernel_channels (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.h34 Eigen::Index kernel_y, Eigen::Index kernel_channels, 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.cc960 int64_t kernel_channels = in HandleConvolution() local