Home
last modified time | relevance | path

Searched defs:out_rows (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.cc157 int filter_cols, int pad_rows, int pad_cols, int out_rows, in Run()
173 int filter_cols, int pad_rows, int pad_cols, int out_rows, in Run()
214 int filter_cols, int pad_rows, int pad_cols, int out_rows, in Run()
228 int filter_cols, int pad_rows, int pad_cols, int out_rows, in Run()
409 int64 out_rows = 0, out_cols = 0; in ComputeConv2DDimension() local
668 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
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.cc245 const int64 out_rows = out_backprop.dim_size(1); in Compute() local
Dconv_ops_3d.cc226 int64 out_rows = GetTensorDim(*output, data_format, '1'); in launch() local
Ddepthwise_conv_op.cc348 int64 out_rows = 0, out_cols = 0, pad_rows = 0, pad_cols = 0; in Compute() local
Dmkl_conv_ops.h390 int64 out_rows = 0, out_cols = 0, out_planes = 0; variable
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_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.cc192 const int64 out_rows = args.out_rows; 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.cc70 int out_rows = 7; in TEST() local