Home
last modified time | relevance | path

Searched defs:stride_c (Results 1 – 14 of 14) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Ddepthwise_conv_op.cc280 const int64 stride_c = GetTensorDim(strides_, data_format_, 'C'); in DepthwiseConv2dNativeOp() local
Dmkl_conv_ops.h561 int stride_c = GetTensorDim(strides_, data_format_, 'C'); in MklConvBackpropCommonOp() local
Ddepthwise_conv_grad_op.cc554 const int64 stride_c = GetTensorDim(strides_, data_format_, 'C'); in DepthwiseConv2dNativeBackpropInputOp() local
1028 const int64 stride_c = GetTensorDim(strides_, data_format_, 'C'); in DepthwiseConv2dNativeBackpropFilterOp() local
Dconv_ops_using_gemm.cc445 const int64 stride_c = GetTensorDim(strides_, data_format_, 'C'); in Conv2DUsingGemmOp() local
Dpooling_ops_3d.cc677 const int32 stride_c = GetTensorDim(stride_, data_format_, 'C'); in MaxPooling3dGradGradOp() local
Dconv_grad_filter_ops.cc454 int stride_c = GetTensorDim(strides_, data_format_, 'C'); in Conv2DSlowBackpropFilterOp() local
Dmkl_conv_ops.cc432 const int64 stride_c = GetTensorDim(strides_, data_format_, 'C'); in MklConvOp() local
876 const int64 stride_c = GetTensorDim(strides_, data_format_, 'C'); in MklConvOp() local
Dconv_grad_input_ops.cc591 int stride_c = GetTensorDim(strides_, data_format_, 'C'); in Conv2DSlowBackpropInputOp() local
Dconv_ops_fused_image_transform.cc637 const int64 stride_c = GetTensorDim(strides_, FORMAT_NHWC, 'C'); in FusedResizeConv2DUsingGemmOp() local
Dconv_ops.cc305 const int64 stride_c = GetTensorDim(strides, data_format, 'C'); in InitConv2DParameters() local
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_blas.cc2277 int64 stride_c, int batch_count) { in DoBlasGemmStridedBatched()
2287 float beta, DeviceMemory<float> *c, int ldc, int64 stride_c, in DoBlasGemmStridedBatched()
2298 double beta, DeviceMemory<double> *c, int ldc, int64 stride_c, in DoBlasGemmStridedBatched()
2311 int64 stride_c, int batch_count) { in DoBlasGemmStridedBatched()
2323 int64 stride_c, int batch_count) { in DoBlasGemmStridedBatched()
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_blas.cc2363 int64 stride_c, int batch_count) { in DoBlasGemmStridedBatched()
2419 float beta, DeviceMemory<float> *c, int ldc, int64 stride_c, in DoBlasGemmStridedBatched()
2432 double beta, DeviceMemory<double> *c, int ldc, int64 stride_c, in DoBlasGemmStridedBatched()
2447 int64 stride_c, int batch_count) { in DoBlasGemmStridedBatched()
2462 int64 stride_c, int batch_count) { in DoBlasGemmStridedBatched()
/external/tensorflow/tensorflow/stream_executor/
Dstream.cc4557 float beta, DeviceMemory<Eigen::half> *c, int ldc, int64 stride_c, in ThenBlasGemmStridedBatched()
4578 float beta, DeviceMemory<float> *c, int ldc, int64 stride_c, in ThenBlasGemmStridedBatched()
4599 double beta, DeviceMemory<double> *c, int ldc, int64 stride_c, in ThenBlasGemmStridedBatched()
4622 int64 stride_c, int batch_count) { in ThenBlasGemmStridedBatched()
4645 int64 stride_c, int batch_count) { in ThenBlasGemmStridedBatched()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Doptimized_ops.h2247 int stride_c = n; in Conv() local