Home
last modified time | relevance | path

Searched defs:in_rows (Results 1 – 16 of 16) 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
Dfused_batch_norm_op.cc123 const int64 in_rows = GetTensorDim(x_input, tensor_format, 'H'); in operator ()() local
261 const int64 in_rows = GetTensorDim(x_input, tensor_format, 'H'); in operator ()() local
353 const int64 in_rows = GetTensorDim(x_input, tensor_format, 'H'); in operator ()() local
1273 int64 in_rows = GetTensorDim(x, tensor_format_, '1'); in ComputeWithReservedSpace() local
1434 int64 in_rows = GetTensorDim(x, tensor_format_, '1'); in ComputeWithReservedSpace() 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.cc298 const int64 in_rows = output_shape.dim_size(1); in Compute() local
562 const int64 in_rows = output_shape.dim_size(1); in Compute() local
Dfractional_avg_pool_op.cc254 const int64 in_rows = orig_input_tensor_shape_flat(1); in Compute() local
Dpooling_ops_common.h236 const int32 in_rows = params.tensor_in_rows; in SpatialMaxPool() local
Dconv_ops_3d.cc224 int64 in_rows = GetTensorDim(input, data_format, '1'); in launch() local
Dmaxpooling_op.cc112 const int32 in_rows = params.tensor_in_rows; in SpatialMaxPoolWithArgMaxHelper() local
592 const int32 in_rows = params.tensor_in_rows; in SpatialMaxPoolGradGrad() local
Dpooling_ops_3d.cc576 const int32 in_rows = params.tensor_in_rows; in launch() local
Dconv_ops.cc667 int64 in_rows = GetTensorDim(input, data_format, 'H'); in operator ()() local
/external/tensorflow/tensorflow/core/kernels/image/
Dextract_volume_patches_op.cc80 const int in_rows = input.dim_size(2); in Compute() local
Dextract_image_patches_op.cc73 const int in_rows = input.dim_size(1); in Compute() local
/external/tensorflow/tensorflow/core/ops/
Dnn_ops.cc1030 auto in_rows = c->Value(in_rows_dim); in __anon4650176f1702() local
Darray_ops.cc2641 auto in_rows = c->Value(in_rows_dim); in __anondb9326b23902() local
2747 auto in_rows = c->Value(in_rows_dim); in __anondb9326b23a02() local