Home
last modified time | relevance | path

Searched defs:col_stride (Results 1 – 25 of 30) sorted by relevance

12

/external/executorch/kernels/portable/cpu/
Dop_tril.cpp46 int64_t col_stride) { in apply_tril()
103 int64_t col_stride = strides_ref[ndim - 1]; in tril_kernel() local
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Druntime_single_threaded_conv2d.cc28 int64_t col_stride, int64_t padding_top, int64_t padding_bottom, in __xla_cpu_runtime_EigenSingleThreadedConv2DF16()
47 int64_t output_cols, int64_t row_stride, int64_t col_stride, in __xla_cpu_runtime_EigenSingleThreadedConv2DF32()
Druntime_conv2d_acl.cc34 int64_t output_cols, int64_t row_stride, int64_t col_stride, in ACLDepthwiseConvImpl()
133 int64_t row_stride, int64_t col_stride, in ACLGemmConvImpl()
238 int64_t output_cols, int64_t row_stride, int64_t col_stride, in __xla_cpu_runtime_ACLConv2DF32()
Druntime_conv2d.cc30 int64_t output_cols, int64_t row_stride, int64_t col_stride, in __xla_cpu_runtime_EigenConv2DF32()
53 int64_t col_stride, int64_t padding_top, int64_t padding_bottom, in __xla_cpu_runtime_EigenConv2DF16()
Druntime_conv2d_mkl.cc58 int64_t col_stride, int64_t padding_top, in MKLConvImpl()
159 int64_t output_cols, int64_t row_stride, int64_t col_stride, in __xla_cpu_runtime_MKLConv2DF32()
Druntime_conv2d_acl.h83 int64_t col_stride, int64_t padding_top, int64_t padding_bottom, in __xla_cpu_runtime_ACLConv2DF32()
/external/libopus/dnn/
Dvec.h50 static inline void sgemv16x1(float *out, const float *weights, int rows, int cols, int col_stride, … in sgemv16x1()
84 static inline void sgemv8x1(float *out, const float *weights, int rows, int cols, int col_stride, c… in sgemv8x1()
110 static inline void sgemv(float *out, const float *weights, int rows, int cols, int col_stride, cons… in sgemv()
Dvec_neon.h208 static inline void sgemv16x1(float *out, const float *weights, int rows, int cols, int col_stride, … in sgemv16x1()
252 static inline void sgemv8x1(float *out, const float *weights, int rows, int cols, int col_stride, c… in sgemv8x1()
287 static inline void sgemv(float *out, const float *weights, int rows, int cols, int col_stride, cons… in sgemv()
/external/tensorflow/tensorflow/core/kernels/
Dpooling_ops_common.h69 int col_stride; member
241 const int32_t col_stride = params.col_stride; in SpatialMaxPool() local
Dconv_ops_fused_impl.h98 int row_stride, int col_stride, // in LaunchFusedConv2DWithOutputKernel()
Dpooling_ops_3d.h63 int col_stride; member
Davgpooling_op.cc324 const int col_stride = stride_[2]; in Compute() local
585 const int col_stride = stride_[2]; in Compute() local
Dconv_grad_input_ops.cc76 int col_dilation, int row_stride, int col_stride, const Padding& padding, in operator ()()
Dmaxpooling_op.cc121 const int32_t col_stride = params.col_stride; in SpatialMaxPoolWithArgMaxHelper() local
623 const int32_t col_stride = params.col_stride; in SpatialMaxPoolGradGrad() local
Dpooling_ops_3d.cc609 const int32_t col_stride = params.col_stride; in launch() local
Dconv_grad_filter_ops.cc672 int col_dilation, int row_stride, int col_stride, const Padding& padding, in operator ()()
Dops_util_test.cc40 int col_stride; member
/external/libopus/src/
Dmlp.c60 static void gemm_accum(float *out, const opus_int8 *weights, int rows, int cols, int col_stride, co… in gemm_accum()
/external/tensorflow/tensorflow/python/kernel_tests/nn_ops/
Dpooling_ops_test.py1157 col_stride, argument
1235 col_stride, argument
1535 window_cols, row_stride, col_stride, padding, v2): argument
1566 window_rows, window_cols, row_stride, col_stride, argument
2121 window_cols, row_stride, col_stride, padding): argument
/external/ruy/ruy/
Dmatrix.h197 int col_stride = layout.order() == Order::kRowMajor ? 1 : layout.stride(); in Offset() local
Dmat.h368 int col_stride = layout.order == Order::kRowMajor ? 1 : layout.stride; in Offset() local
/external/eigen/unsupported/test/
Dcxx11_tensor_image_patch.cpp418 int col_stride = 1; in test_patch_padding_same_negative_padding_clip_to_zero() local
/external/swiftshader/third_party/SPIRV-Tools/source/opt/
Dinst_bindless_check_pass.cpp483 uint32_t col_stride = ByteSize(comp_ty_id, 0u, false, false); in GenLastByteIdx() local
/external/eigen/unsupported/Eigen/CXX11/src/Tensor/
DTensorBase.h844 const Index row_stride, const Index col_stride, in extract_image_patches()
/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_pooling_ops_common.h381 int col_stride; member

12