| /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
|