Home
last modified time | relevance | path

Searched defs:out_cols (Results 1 – 19 of 19) 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.cc241 int out_cols, int out_depth, int dilation_rows, in Run()
282 int out_cols, int out_depth, int stride_rows, int stride_cols, in Run()
296 int out_cols, int out_depth, int dilation_rows, in Run()
476 int64 out_rows = 0, out_cols = 0; in ComputeConv2DDimension() local
760 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
Dfractional_avg_pool_op.cc246 const int64 out_cols = out_backprop.dim_size(2); in Compute() local
Ddepthwise_conv_op.cc374 int64 out_rows = 0, out_cols = 0, pad_top = 0, pad_bottom = 0, pad_left = 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()
Dconv_ops_3d.cc237 int64 out_cols = GetTensorDim(*output, data_format, '2'); in launch() local
Dconv_ops_using_gemm.cc522 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.cc128 int64 out_rows = 0, out_cols = 0, pad_rows = 0, pad_cols = 0; in BM_ConvFloat() local
537 int64 out_rows = 0, out_cols = 0, pad_rows = 0, pad_cols = 0; in BM_ConvFloatDepthwise() local
Dconv_ops_fused_image_transform.cc822 int64 out_rows = 0, out_cols = 0, pad_rows = 0, pad_cols = 0; in Compute() local
Ddepthwise_conv_grad_op.cc205 const int64 out_cols = args.out_cols; in CopyOutputBackpropRegion() local
/external/tensorflow/tensorflow/core/kernels/image/
Dextract_volume_patches_op.cc112 int64 out_planes = 0, out_rows = 0, out_cols = 0; in Compute() local
Dextract_image_patches_op.cc89 int64 out_rows = 0, out_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
/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_conv_ops.h396 int64 out_rows = 0, out_cols = 0, out_planes = 0; variable
/external/tensorflow/tensorflow/core/util/sparse/
Dsparse_tensor.h381 const int64 out_cols = out_shape.dim_size(1); in ToDense() local