Home
last modified time | relevance | path

Searched refs:stride_c (Results 1 – 17 of 17) sorted by relevance

/external/tensorflow/tensorflow/python/framework/
Dcommon_shapes.py214 stride_b, stride_d, stride_r, stride_c = op.get_attr("strides")
216 stride_b, stride_r, stride_c, stride_d = op.get_attr("strides")
226 filter_cols, stride_r, stride_c,
272 stride_b, stride_r, stride_c, stride_d = op.get_attr("strides")
276 if stride_r != stride_c:
335 stride_b, stride_r, stride_c, stride_d = op.get_attr("strides")
339 if stride_r != stride_c:
390 stride_b, stride_d, stride_r, stride_c = op.get_attr("strides")
393 stride_b, stride_r, stride_c, stride_d = op.get_attr("strides")
413 ksize_c, stride_r, stride_c,
[all …]
/external/tensorflow/tensorflow/core/kernels/
Dconv_ops_using_gemm.cc445 const int64 stride_c = GetTensorDim(strides_, data_format_, 'C'); in Conv2DUsingGemmOp() local
447 context, stride_n == 1 && stride_c == 1, in Conv2DUsingGemmOp()
Ddepthwise_conv_op.cc280 const int64 stride_c = GetTensorDim(strides_, data_format_, 'C'); in DepthwiseConv2dNativeOp() local
287 context, (stride_n == 1 && stride_c == 1), in DepthwiseConv2dNativeOp()
Ddepthwise_conv_grad_op.cc554 const int64 stride_c = GetTensorDim(strides_, data_format_, 'C'); in DepthwiseConv2dNativeBackpropInputOp() local
561 context, (stride_n == 1 && stride_c == 1), in DepthwiseConv2dNativeBackpropInputOp()
1028 const int64 stride_c = GetTensorDim(strides_, data_format_, 'C'); in DepthwiseConv2dNativeBackpropFilterOp() local
1035 context, (stride_n == 1 && stride_c == 1), in DepthwiseConv2dNativeBackpropFilterOp()
Dmkl_conv_ops.h561 int stride_c = GetTensorDim(strides_, data_format_, 'C'); in MklConvBackpropCommonOp() local
565 context, (stride_n == 1 && stride_c == 1), in MklConvBackpropCommonOp()
Dconv_ops_fused_image_transform.cc637 const int64 stride_c = GetTensorDim(strides_, FORMAT_NHWC, 'C'); in FusedResizeConv2DUsingGemmOp() local
639 context, stride_n == 1 && stride_c == 1, in FusedResizeConv2DUsingGemmOp()
Dmkl_conv_ops.cc432 const int64 stride_c = GetTensorDim(strides_, data_format_, 'C'); in MklConvOp() local
434 context, stride_n == 1 && stride_c == 1, in MklConvOp()
876 const int64 stride_c = GetTensorDim(strides_, data_format_, 'C'); in MklConvOp() local
878 context, stride_n == 1 && stride_c == 1, in MklConvOp()
Dconv_ops.cc305 const int64 stride_c = GetTensorDim(strides, data_format, 'C'); in InitConv2DParameters() local
309 stride_n == 1 && stride_c == 1, in InitConv2DParameters()
Dpooling_ops_3d.cc677 const int32 stride_c = GetTensorDim(stride_, data_format_, 'C'); in MaxPooling3dGradGradOp() local
678 OP_REQUIRES(context, ksize_c == 1 && stride_c == 1, in MaxPooling3dGradGradOp()
Dconv_grad_filter_ops.cc454 int stride_c = GetTensorDim(strides_, data_format_, 'C'); in Conv2DSlowBackpropFilterOp() local
458 context, (stride_n == 1 && stride_c == 1), in Conv2DSlowBackpropFilterOp()
Dconv_grad_input_ops.cc591 int stride_c = GetTensorDim(strides_, data_format_, 'C'); in Conv2DSlowBackpropInputOp() local
595 context, (stride_n == 1 && stride_c == 1), in Conv2DSlowBackpropInputOp()
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_blas.cc2363 int64 stride_c, int batch_count) { in DoBlasGemmStridedBatched() argument
2382 ldb, stride_b, &beta, GpuMemoryMutable(c), CUDA_R_16F, ldc, stride_c, in DoBlasGemmStridedBatched()
2400 reinterpret_cast<__half *>(GpuMemoryMutable(c) + batch * stride_c); in DoBlasGemmStridedBatched()
2419 float beta, DeviceMemory<float> *c, int ldc, int64 stride_c, in DoBlasGemmStridedBatched() argument
2425 GpuMemoryMutable(c), ldc, stride_c, batch_count); in DoBlasGemmStridedBatched()
2432 double beta, DeviceMemory<double> *c, int ldc, int64 stride_c, in DoBlasGemmStridedBatched() argument
2438 GpuMemoryMutable(c), ldc, stride_c, batch_count); in DoBlasGemmStridedBatched()
2447 int64 stride_c, int batch_count) { in DoBlasGemmStridedBatched() argument
2453 GpuComplex(GpuMemoryMutable(c)), ldc, stride_c, batch_count); in DoBlasGemmStridedBatched()
2462 int64 stride_c, int batch_count) { in DoBlasGemmStridedBatched() argument
[all …]
/external/tensorflow/tensorflow/stream_executor/
Dblas.h1131 int64 stride_c, int batch_count) = 0;
1136 float beta, DeviceMemory<float> *c, int ldc, int64 stride_c,
1142 double beta, DeviceMemory<double> *c, int ldc, int64 stride_c,
1150 int64 stride_c, int batch_count) = 0;
1157 int64 stride_c, int batch_count) = 0;
2033 DeviceMemory<Eigen::half> *c, int ldc, int64 stride_c, int batch_count); \
2039 int64 stride_c, int batch_count); \
2045 DeviceMemory<double> *c, int ldc, int64 stride_c, int batch_count); \
2052 int64 stride_c, int batch_count); \
2059 int ldc, int64 stride_c, int batch_count); \
Dstream.cc4557 float beta, DeviceMemory<Eigen::half> *c, int ldc, int64 stride_c, in ThenBlasGemmStridedBatched() argument
4562 PARAM(stride_c), PARAM(batch_count)); in ThenBlasGemmStridedBatched()
4571 c, ldc, stride_c, batch_count); in ThenBlasGemmStridedBatched()
4578 float beta, DeviceMemory<float> *c, int ldc, int64 stride_c, in ThenBlasGemmStridedBatched() argument
4583 PARAM(stride_c), PARAM(batch_count)); in ThenBlasGemmStridedBatched()
4592 c, ldc, stride_c, batch_count); in ThenBlasGemmStridedBatched()
4599 double beta, DeviceMemory<double> *c, int ldc, int64 stride_c, in ThenBlasGemmStridedBatched() argument
4604 PARAM(stride_c), PARAM(batch_count)); in ThenBlasGemmStridedBatched()
4613 c, ldc, stride_c, batch_count); in ThenBlasGemmStridedBatched()
4622 int64 stride_c, int batch_count) { in ThenBlasGemmStridedBatched() argument
[all …]
Dstream.h1455 int64 stride_c, int batch_count);
1460 float beta, DeviceMemory<float> *c, int ldc, int64 stride_c,
1466 double beta, DeviceMemory<double> *c, int ldc, int64 stride_c,
1474 int64 stride_c, int batch_count);
1481 int64 stride_c, int batch_count);
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_blas.cc2277 int64 stride_c, int batch_count) { in DoBlasGemmStridedBatched() argument
2287 float beta, DeviceMemory<float> *c, int ldc, int64 stride_c, in DoBlasGemmStridedBatched() argument
2298 double beta, DeviceMemory<double> *c, int ldc, int64 stride_c, in DoBlasGemmStridedBatched() argument
2311 int64 stride_c, int batch_count) { in DoBlasGemmStridedBatched() argument
2323 int64 stride_c, int batch_count) { in DoBlasGemmStridedBatched() argument
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Doptimized_ops.h2247 int stride_c = n; in Conv() local
2250 stride_a, b, stride_b, 0.0f, c, stride_c); in Conv()