/external/tensorflow/tensorflow/core/kernels/ |
dilation_ops.cc | 67 int* rate_cols, int64* pad_top, int64* pad_left, 108 padding, out_rows, pad_top)); 128 int64 pad_top = 0, pad_left = 0; variable 131 &rate_rows, &rate_cols, &pad_top, &pad_left, &out_rows, 152 pad_top, pad_left, output->tensor<T, 4>()); 166 int stride_cols, int rate_rows, int rate_cols, int pad_top, 183 int h_beg = h_out * stride_rows - pad_top; 227 int64 pad_top = 0, pad_left = 0; variable 230 &rate_rows, &rate_cols, &pad_top, &pad_left, &out_rows, 258 stride_cols, rate_rows, rate_cols, pad_top, pad_left 347 int64 pad_top = 0, pad_left = 0; variable [all...] |
ops_util_test.cc | 45 int pad_top; member in struct:tensorflow::__anon45438::OpsUtilTest::padding_struct::__anon45440 99 EXPECT_EQ(pad_struct.output.pad_top, pad_rows); 105 int64 new_height, new_width, pad_top, pad_bottom, pad_left, pad_right; local 109 &pad_top, &pad_bottom); 118 EXPECT_EQ(pad_struct.output.pad_top, pad_top);
|
mkl_conv_ops.h | 391 int64 pad_top = 0, pad_bottom = 0, pad_left, pad_right; local 398 pad_top = static_cast<int64>((*pad_l)[0]); 408 padding_type, &out_rows, &pad_top, &pad_bottom)); 419 padding_, &out_rows, &pad_top, &pad_bottom)); 431 *pad_l = {static_cast<int>(pad_top), static_cast<int>(pad_left)}; 436 *pad_l = {static_cast<int>(pad_D1), static_cast<int>(pad_top),
|
conv_grad_filter_ops.cc | 265 int64 pad_top, pad_bottom; variable 272 &dims.spatial_dims[0].output_size, &pad_top, &pad_bottom)); 281 if (pad_left == pad_right && pad_top == pad_bottom) { 288 static_cast<int>(pad_top), static_cast<int>(pad_left), 368 auto shard = [&input_data, &col_buffer_data, &dims, &pad_top, &pad_left, 380 dims.spatial_dims[1].filter_size, pad_top, pad_left, pad_bottom, [all...] |
conv_grad_input_ops.cc | 346 int64 pad_top, pad_bottom; variable 353 &dims.spatial_dims[0].output_size, &pad_top, &pad_bottom)); 361 if (pad_left == pad_right && pad_top == pad_bottom) { 369 static_cast<int>(pad_top), static_cast<int>(pad_left), 375 int64 pad_top, pad_bottom; variable 383 &dims.spatial_dims[0].output_size, &pad_top, &pad_bottom)); 490 dims.spatial_dims[1].filter_size, pad_top, pad_left, pad_bottom, 503 auto shard = [&context, &dims, &pad_top, &pad_left, &pad_bottom, 521 dims.spatial_dims[1].filter_size, pad_top, pad_left, [all...] |
mkl_pooling_ops_common.h | 391 int64 pad_top; member in struct:tensorflow::MklPoolParameters 418 pad_top(0), 531 *padding_left = memory::dims({static_cast<int>(pool_params->pad_top), 545 static_cast<int>(pool_params->pad_top),
|
mkl_conv_ops.cc | 1193 int64 pad_top, pad_left; local [all...] |