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.cc289 const int64_t stride_c = GetTensorDim(strides_, data_format_, 'C'); in DepthwiseConv2dNativeOp() local
Ddepthwise_conv_grad_op.cc571 const int64_t stride_c = GetTensorDim(strides_, data_format_, 'C'); in DepthwiseConv2dNativeBackpropInputOp() local
1074 const int64_t stride_c = GetTensorDim(strides_, data_format_, 'C'); in DepthwiseConv2dNativeBackpropFilterOp() local
Dconv_ops_using_gemm.cc447 const int64_t stride_c = GetTensorDim(strides_, data_format_, 'C'); in Conv2DUsingGemmOp() local
Dpooling_ops_3d.cc696 const int32_t stride_c = GetTensorDim(stride_, data_format_, 'C'); in MaxPooling3dGradGradOp() local
Dconv_grad_filter_ops.cc275 int stride_c = GetTensorDim(strides_, data_format_, 'C'); in Conv2DBackpropFilterOp() local
Dconv_ops_fused_image_transform.cc639 const int64_t stride_c = GetTensorDim(strides_, FORMAT_NHWC, 'C'); in FusedResizeConv2DUsingGemmOp() local
Dconv_ops.cc517 const int64_t stride_c = GetTensorDim(strides, data_format, 'C'); in InitConv2DParameters() local
/external/tensorflow/tensorflow/compiler/xla/stream_executor/
Dstream.h923 int64_t stride_c, int batch_count, blas::ComputationType computation_type, in ThenBlasGemmStridedBatchedWithAlgorithm()
1038 int64_t stride_c, int batch_count) { in ThenBlasGemmStridedBatched()
/external/tensorflow/tensorflow/core/common_runtime/
Deval_const_tensor.cc139 InferenceContext* stride_c = refiner.GetContext(stride_node); in TryToInferTensorOutputFromStridedSliceNode() local
/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_conv_ops.h625 int stride_c = GetTensorDim(strides_, data_format_, 'C'); in MklConvBackpropCommonOp() local
Dmkl_conv_ops.cc576 const int64 stride_c = GetTensorDim(strides_, data_format_, 'C'); in MklConvOp() local
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_blas.cc645 DeviceMemoryBase *c, blas::DataType type_c, int ldc, int64_t stride_c, in DoBlasGemmStridedBatchedWithAlgorithm()
1074 DeviceMemoryBase *c, int ldc, int64_t stride_c, int batch_count) { in DoBlasGemmStridedBatched()
/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/
Dcuda_blas.cc1049 DeviceMemoryBase *c, blas::DataType type_c, int ldc, int64_t stride_c, in DoBlasInternalImpl()
1412 DeviceMemoryBase *c, int ldc, int64_t stride_c, int batch_count) { in DoBlasInternalImpl()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Dlegacy_optimized_ops.h2490 int stride_c = n; in Conv() local
Doptimized_ops.h970 int stride_c = n; in Conv() local