Home
last modified time | relevance | path

Searched defs:out_rows (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.cc224 int filter_cols, int pad_rows, int pad_cols, int out_rows, in Run()
240 int filter_cols, int pad_rows, int pad_cols, int out_rows, in Run()
281 int filter_cols, int pad_rows, int pad_cols, int out_rows, in Run()
295 int filter_cols, int pad_rows, int pad_cols, int out_rows, in Run()
476 int64 out_rows = 0, out_cols = 0; in ComputeConv2DDimension() local
759 const int64 out_rows = GetTensorDim(*output, data_format, 'H'); in operator ()() local
Ddepthwise_conv_op.h39 int out_rows; member
Ddeep_conv2d.h79 int out_rows; member
Dfractional_avg_pool_op.cc245 const int64 out_rows = out_backprop.dim_size(1); 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.cc236 int64 out_rows = GetTensorDim(*output, data_format, '1'); 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.cc204 const int64 out_rows = args.out_rows; 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.cc70 int out_rows = 7; 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.h380 const int64 out_rows = out_shape.dim_size(0); in ToDense() local