Home
last modified time | relevance | path

Searched defs:out_cols (Results 1 – 20 of 20) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Ddilation_ops.cc68 int64* out_rows, int64* out_cols) { in ParseSizes()
129 int64 out_rows = 0, out_cols = 0; in Compute() local
228 int64 out_rows = 0, out_cols = 0; in Compute() local
348 int64 out_rows = 0, out_cols = 0; in Compute() local
Dconv_ops.cc174 int out_cols, int out_depth, int dilation_rows, in Run()
215 int out_cols, int out_depth, int stride_rows, int stride_cols, in Run()
229 int out_cols, int out_depth, int dilation_rows, in Run()
409 int64 out_rows = 0, out_cols = 0; in ComputeConv2DDimension() local
669 const int64 out_cols = GetTensorDim(*output, data_format, 'W'); in operator ()() local
Ddepthwise_conv_op.h40 int out_cols; member
Ddeep_conv2d.h80 int out_cols; member
Dextract_image_patches_op.cc86 int64 out_rows = 0, out_cols = 0; in Compute() local
Dextract_volume_patches_op.cc109 int64 out_planes = 0, out_rows = 0, out_cols = 0; in Compute() local
Dpooling_ops_3d_sycl.h126 const int out_rows, const int out_cols, in MaxPool3DSYCL()
Dfractional_avg_pool_op.cc246 const int64 out_cols = out_backprop.dim_size(2); in Compute() local
Dconv_ops_3d.cc227 int64 out_cols = GetTensorDim(*output, data_format, '2'); in launch() local
Ddepthwise_conv_op.cc348 int64 out_rows = 0, out_cols = 0, pad_rows = 0, pad_cols = 0; in Compute() local
Ddeep_conv2d.cc50 int out_depth, int out_rows, int out_cols) { in GetDeepConvCost()
75 int out_depth, int out_rows, int out_cols) { in GetDirectConvCost()
99 int out_rows, int out_cols) { in CanUseDeepConv2D()
Dmkl_conv_ops.h390 int64 out_rows = 0, out_cols = 0, out_planes = 0; variable
Dconv_ops_using_gemm.cc520 int64 out_rows = 0, out_cols = 0, pad_rows = 0, pad_cols = 0; in Compute() local
Dquantized_conv_ops.cc540 int64 out_rows = 0, out_cols = 0, pad_rows = 0, pad_cols = 0; in Compute() local
Dnn_ops_test.cc126 int64 out_rows = 0, out_cols = 0, pad_rows = 0, pad_cols = 0; in BM_ConvFloat() local
520 int64 out_rows = 0, out_cols = 0, pad_rows = 0, pad_cols = 0; in BM_ConvFloatDepthwise() local
Dconv_ops_fused_image_transform.cc825 int64 out_rows = 0, out_cols = 0, pad_rows = 0, pad_cols = 0; in Compute() local
Ddepthwise_conv_grad_op.cc193 const int64 out_cols = args.out_cols; in CopyOutputBackpropRegion() local
Dmkl_conv_ops.cc525 int64 out_rows = 0, out_cols = 0, pad_rows = 0, pad_cols = 0; in Compute() local
/external/tensorflow/tensorflow/core/kernels/neon/
Dneon_depthwise_conv_op.cc90 int64 out_rows = 0, out_cols = 0, pad_rows = 0, pad_cols = 0; in Compute() local
/external/tensorflow/tensorflow/core/grappler/costs/
Dutils_test.cc71 int out_cols = 9; in TEST() local