/external/tensorflow/tensorflow/stream_executor/ |
D | blas.h | 231 int64 stride_c = 0; member
|
D | stream.cc | 4220 float beta, DeviceMemory<Eigen::half> *c, int ldc, int64 stride_c, in ThenBlasGemmStridedBatched() 4241 float beta, DeviceMemory<float> *c, int ldc, int64 stride_c, in ThenBlasGemmStridedBatched() 4262 double beta, DeviceMemory<double> *c, int ldc, int64 stride_c, in ThenBlasGemmStridedBatched() 4285 int64 stride_c, int batch_count) { in ThenBlasGemmStridedBatched() 4308 int64 stride_c, int batch_count) { in ThenBlasGemmStridedBatched()
|
/external/tensorflow/tensorflow/core/kernels/ |
D | depthwise_conv_op.cc | 285 const int64 stride_c = GetTensorDim(strides_, data_format_, 'C'); in DepthwiseConv2dNativeOp() local
|
D | depthwise_conv_grad_op.cc | 566 const int64 stride_c = GetTensorDim(strides_, data_format_, 'C'); in DepthwiseConv2dNativeBackpropInputOp() local 1069 const int64 stride_c = GetTensorDim(strides_, data_format_, 'C'); in DepthwiseConv2dNativeBackpropFilterOp() local
|
D | conv_ops_using_gemm.cc | 447 const int64 stride_c = GetTensorDim(strides_, data_format_, 'C'); in Conv2DUsingGemmOp() local
|
D | pooling_ops_3d.cc | 673 const int32 stride_c = GetTensorDim(stride_, data_format_, 'C'); in MaxPooling3dGradGradOp() local
|
D | conv_ops_fused_image_transform.cc | 638 const int64 stride_c = GetTensorDim(strides_, FORMAT_NHWC, 'C'); in FusedResizeConv2DUsingGemmOp() local
|
D | conv_grad_filter_ops.cc | 271 int stride_c = GetTensorDim(strides_, data_format_, 'C'); in Conv2DBackpropFilterOp() local
|
D | conv_ops.cc | 372 const int64 stride_c = GetTensorDim(strides, data_format, 'C'); in InitConv2DParameters() local
|
/external/tensorflow/tensorflow/core/kernels/mkl/ |
D | mkl_conv_ops.h | 570 int stride_c = GetTensorDim(strides_, data_format_, 'C'); in MklConvBackpropCommonOp() local
|
D | mkl_conv_ops.cc | 450 const int64 stride_c = GetTensorDim(strides_, data_format_, 'C'); in MklConvOp() local
|
/external/tensorflow/tensorflow/stream_executor/rocm/ |
D | rocm_blas.cc | 2566 int64 stride_c, int batch_count) { in DoBlasGemmStridedBatched() 2587 float beta, DeviceMemory<float> *c, int ldc, int64 stride_c, in DoBlasGemmStridedBatched() 2606 double beta, DeviceMemory<double> *c, int ldc, int64 stride_c, in DoBlasGemmStridedBatched() 2627 int64 stride_c, int batch_count) { in DoBlasGemmStridedBatched() 2642 int64 stride_c, int batch_count) { in DoBlasGemmStridedBatched()
|
/external/tensorflow/tensorflow/stream_executor/cuda/ |
D | cuda_blas.cc | 2604 int64 stride_c, int batch_count) { in DoBlasInternalImpl() 2657 float beta, DeviceMemory<float> *c, int ldc, int64 stride_c, in DoBlasInternalImpl() 2676 double beta, DeviceMemory<double> *c, int ldc, int64 stride_c, in DoBlasInternalImpl() 2691 int64 stride_c, int batch_count) { in DoBlasInternalImpl() 2708 int64 stride_c, int batch_count) { in DoBlasInternalImpl()
|
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
D | legacy_optimized_ops.h | 2489 int stride_c = n; in Conv() local
|
D | optimized_ops.h | 1301 int stride_c = n; in Conv() local
|