Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Ddilation_ops.cc69 int64_t* out_rows, int64_t* out_cols) { in ParseSizes()
130 int64_t out_rows = 0, out_cols = 0; in Compute() local
235 int64_t out_rows = 0, out_cols = 0; in Compute() local
363 int64_t out_rows = 0, out_cols = 0; in Compute() local
Ddepthwise_conv_op.h40 int out_cols; member
Dconv_ops.cc386 int out_cols, int out_depth, int dilation_rows, in Run()
427 int out_cols, int out_depth, int stride_rows, int stride_cols, in Run()
441 int out_cols, int out_depth, int dilation_rows, in Run()
624 int64_t out_rows = 0, out_cols = 0; in ComputeConv2DDimension() local
911 const int64_t out_cols = GetTensorDim(*output, data_format, 'W'); in operator ()() local
Ddeep_conv2d.h80 int out_cols; member
Dfractional_avg_pool_op.cc276 const int64_t out_cols = out_backprop.dim_size(2); in Compute() local
Ddepthwise_conv_op.cc395 int64_t out_rows = 0, out_cols = 0, pad_top = 0, pad_bottom = 0, in Compute() local
Dconv_ops_3d.cc255 int64_t out_cols = GetTensorDim(*output, data_format, '2'); in launch() local
Ddeep_conv2d.cc51 int out_cols) { in GetDeepConvCost()
76 int out_depth, int out_rows, int out_cols) { in GetDirectConvCost()
100 int out_rows, int out_cols) { in CanUseDeepConv2D()
Dconv_ops_using_gemm.cc522 int64_t out_rows = 0, out_cols = 0, pad_rows = 0, pad_cols = 0; in Compute() local
Dquantized_conv_ops.cc565 int64_t out_rows = 0, out_cols = 0, pad_rows = 0, pad_cols = 0; in Compute() local
Dnn_ops_test.cc132 int64_t out_rows = 0, out_cols = 0, pad_rows = 0, pad_cols = 0; in BM_ConvFloat() local
543 int64_t out_rows = 0, out_cols = 0, pad_rows = 0, pad_cols = 0; in BM_ConvFloatDepthwise() local
Dconv_ops_fused_image_transform.cc823 int64_t out_rows = 0, out_cols = 0, pad_rows = 0, pad_cols = 0; in Compute() local
Ddepthwise_conv_grad_op.cc206 const int64_t out_cols = args.out_cols; in CopyOutputBackpropRegion() local
/external/tensorflow/tensorflow/core/kernels/image/
Dextract_volume_patches_op.cc112 int64_t out_planes = 0, out_rows = 0, out_cols = 0; in Compute() local
Dextract_image_patches_op.cc89 int64_t out_rows = 0, out_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.h424 int64 out_rows = 0, out_cols = 0, out_planes = 0; variable
/external/tensorflow/tensorflow/core/util/sparse/
Dsparse_tensor.h382 const int64_t out_cols = out_shape.dim_size(1); in ToDense() local