Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Ddepthwise_conv_op.cc284 const int64 stride_c = GetTensorDim(strides_, data_format_, 'C'); in DepthwiseConv2dNativeOp() local
Dmkl_conv_ops.h574 int stride_c = GetTensorDim(strides_, data_format_, 'C'); in MklConvBackpropCommonOp() local
Ddepthwise_conv_grad_op.cc558 const int64 stride_c = GetTensorDim(strides_, data_format_, 'C'); in DepthwiseConv2dNativeBackpropInputOp() local
1047 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_ops_fused_image_transform.cc637 const int64 stride_c = GetTensorDim(strides_, FORMAT_NHWC, 'C'); in FusedResizeConv2DUsingGemmOp() local
Dconv_grad_filter_ops.cc271 int stride_c = GetTensorDim(strides_, data_format_, 'C'); in Conv2DBackpropFilterOp() local
Dconv_ops.cc371 const int64 stride_c = GetTensorDim(strides, data_format, 'C'); in InitConv2DParameters() local
Dconv_grad_input_ops.cc396 int stride_c = GetTensorDim(strides_, data_format_, 'C'); in Conv2DBackpropInputOp() local
Dmkl_conv_ops.cc465 const int64 stride_c = GetTensorDim(strides_, data_format_, 'C'); in MklConvOp() local
/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/rocm/
Drocm_blas.cc2344 int64 stride_c, int batch_count) { in DoBlasGemmStridedBatched()
2364 float beta, DeviceMemory<float> *c, int ldc, int64 stride_c, in DoBlasGemmStridedBatched()
2377 double beta, DeviceMemory<double> *c, int ldc, int64 stride_c, in DoBlasGemmStridedBatched()
2392 int64 stride_c, int batch_count) { in DoBlasGemmStridedBatched()
2404 int64 stride_c, int batch_count) { in DoBlasGemmStridedBatched()
/external/tensorflow/tensorflow/stream_executor/
Dstream.cc4699 float beta, DeviceMemory<Eigen::half> *c, int ldc, int64 stride_c, in ThenBlasGemmStridedBatched()
4720 float beta, DeviceMemory<float> *c, int ldc, int64 stride_c, in ThenBlasGemmStridedBatched()
4741 double beta, DeviceMemory<double> *c, int ldc, int64 stride_c, in ThenBlasGemmStridedBatched()
4764 int64 stride_c, int batch_count) { in ThenBlasGemmStridedBatched()
4787 int64 stride_c, int batch_count) { in ThenBlasGemmStridedBatched()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Dlegacy_optimized_ops.h2488 int stride_c = n; in Conv() local
Doptimized_ops.h1211 int stride_c = n; in Conv() local