Searched defs:kernel_z (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | runtime_conv3d.cc | 29 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()
|
| D | runtime_single_threaded_conv3d.cc | 26 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()
|
| D | runtime_conv_impl.h | 111 Eigen::Index kernel_x, Eigen::Index kernel_y, Eigen::Index kernel_z, in EigenConv3DImpl()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/ |
| D | depthwise_conv.h | 191 const int kernel_z = weights.shape.d; in RearrangeWeightsForDWConv3D() local
223 const int kernel_z = weights.shape.d; in UploadWeightsForDWConv3D() local
|