/external/tensorflow/tensorflow/compiler/xla/tests/ |
D | pad_test.cc | 343 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/ |
D | depthwise_conv_op.h | 28 int in_rows; member
|
D | deep_conv2d.h | 70 int in_rows; member
|
D | fused_batch_norm_op.cc | 123 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
|
D | eigen_spatial_convolutions_test.cc | 670 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
|
D | avgpooling_op.cc | 298 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
|
D | fractional_avg_pool_op.cc | 254 const int64 in_rows = orig_input_tensor_shape_flat(1); in Compute() local
|
D | pooling_ops_common.h | 236 const int32 in_rows = params.tensor_in_rows; in SpatialMaxPool() local
|
D | conv_ops_3d.cc | 224 int64 in_rows = GetTensorDim(input, data_format, '1'); in launch() local
|
D | maxpooling_op.cc | 112 const int32 in_rows = params.tensor_in_rows; in SpatialMaxPoolWithArgMaxHelper() local 592 const int32 in_rows = params.tensor_in_rows; in SpatialMaxPoolGradGrad() local
|
D | pooling_ops_3d.cc | 576 const int32 in_rows = params.tensor_in_rows; in launch() local
|
D | conv_ops.cc | 667 int64 in_rows = GetTensorDim(input, data_format, 'H'); in operator ()() local
|
/external/tensorflow/tensorflow/core/kernels/image/ |
D | extract_volume_patches_op.cc | 80 const int in_rows = input.dim_size(2); in Compute() local
|
D | extract_image_patches_op.cc | 73 const int in_rows = input.dim_size(1); in Compute() local
|
/external/tensorflow/tensorflow/core/ops/ |
D | nn_ops.cc | 1030 auto in_rows = c->Value(in_rows_dim); in __anon4650176f1702() local
|
D | array_ops.cc | 2641 auto in_rows = c->Value(in_rows_dim); in __anondb9326b23902() local 2747 auto in_rows = c->Value(in_rows_dim); in __anondb9326b23a02() local
|