Home
last modified time | relevance | path

Searched defs:in_rows (Results 1 – 17 of 17) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/tests/
Dpad_test.cc343 constexpr int64 in_rows = 35; in XLA_TEST_P() local
365 constexpr int64 in_rows = 129; in XLA_TEST_P() local
392 constexpr int64 in_rows = 129; in XLA_TEST_P() local
420 constexpr int64 in_rows = 8; in XLA_TEST_P() local
/external/tensorflow/tensorflow/core/kernels/
Ddepthwise_conv_op.h28 int in_rows; member
Ddeep_conv2d.h70 int in_rows; member
Dextract_image_patches_op.cc70 const int in_rows = input.dim_size(1); in Compute() local
Dextract_volume_patches_op.cc77 const int in_rows = input.dim_size(2); in Compute() local
Deigen_spatial_convolutions_test.cc670 const int in_rows = 8; in TEST() local
731 const int in_rows = 8; in TEST() local
793 const int in_rows = 5; in TEST() local
845 const int in_rows = 5; in TEST() local
899 const int in_rows = 8; in TEST() local
965 const int in_rows = 8; in TEST() local
1031 const int in_rows = 7; in TEST() local
1086 const int in_rows = 7; in TEST() local
1142 const int in_rows = 7; in TEST() local
1212 const int in_rows = 7; in TEST() local
Davgpooling_op.cc264 const int64 in_rows = output_shape.dim_size(1); in Compute() local
505 const int64 in_rows = output_shape.dim_size(1); in Compute() local
Dpooling_ops_3d_sycl.h125 const int in_rows, const int in_cols, const int out_planes, in MaxPool3DSYCL()
Dpooling_ops_common.h198 const int32 in_rows = params.tensor_in_rows; in SpatialMaxPool() local
Dfractional_avg_pool_op.cc254 const int64 in_rows = orig_input_tensor_shape_flat(1); in Compute() local
Dconv_ops_3d.cc215 int64 in_rows = GetTensorDim(input, data_format, '1'); in launch() local
Dmaxpooling_op.cc104 const int32 in_rows = params.tensor_in_rows; in SpatialMaxPoolWithArgMaxHelper() local
584 const int32 in_rows = params.tensor_in_rows; in SpatialMaxPoolGradGrad() local
Dpooling_ops_3d.cc580 const int32 in_rows = params.tensor_in_rows; in launch() local
Dconv_ops.cc599 int64 in_rows = GetTensorDim(input, data_format, 'H'); in operator ()() local
/external/tensorflow/tensorflow/core/ops/
Darray_ops.cc2482 auto in_rows = c->Value(in_rows_dim); in __anon7c94107b3902() local
2588 auto in_rows = c->Value(in_rows_dim); in __anon7c94107b3a02() local
Dnn_ops.cc936 auto in_rows = c->Value(in_rows_dim); in __anon3e672dd81d02() local
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.cc2610 int64 in_rows = input_desc.height(); in ShouldIncludeWinogradNonfusedAlgo() local