Home
last modified time | relevance | path

Searched refs:num_block_dims (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/compiler/tests/
Dspacetobatch_op_test.py45 num_block_dims = len(block_shape)
50 (input_array.ndim - 1 - num_block_dims)),
59 reshaped_padded_shape.extend(input_array.shape[num_block_dims + 1:])
60 output_shape.extend(input_array.shape[num_block_dims + 1:])
64 list(np.arange(num_block_dims) * 2 + 2) + [0] +
65 list(np.arange(num_block_dims) * 2 + 1) + list(
66 np.arange(input_array.ndim - num_block_dims - 1) + 1 + num_block_dims
Drandomized_tests.cc1230 const int num_block_dims = 2; in TEST_F() local
1232 RandomDims(num_block_dims, num_block_dims, 0, 5); in TEST_F()
1235 std::vector<int64> input_dims(1 + num_block_dims + 1); in TEST_F()
1237 for (int i = 0; i < num_block_dims; ++i) { in TEST_F()
1241 input_dims[1 + num_block_dims] = RandomDim(); in TEST_F()
1245 for (int i = 0; i < num_block_dims; ++i) { in TEST_F()
1252 TensorShape({num_block_dims, 2}))); in TEST_F()
1266 int num_block_dims = block_dims.size(); in TEST_F() local
1271 std::vector<int64> input_dims(1 + num_block_dims + remaining_dims.size()); in TEST_F()
1273 for (int i = 0; i < num_block_dims; ++i) { in TEST_F()
[all …]
/external/tensorflow/tensorflow/python/kernel_tests/
Dspacetobatch_op_test.py49 num_block_dims = len(block_shape)
54 (input_array.ndim - 1 - num_block_dims)),
63 reshaped_padded_shape.extend(input_array.shape[num_block_dims + 1:])
64 output_shape.extend(input_array.shape[num_block_dims + 1:])
68 list(np.arange(num_block_dims) * 2 + 2) + [0] +
69 list(np.arange(num_block_dims) * 2 + 1) + list(
70 np.arange(input_array.ndim - num_block_dims - 1) + 1 + num_block_dims
632 num_block_dims = len(block_shape)
633 self.assertEqual(len(input_shape), num_block_dims)
635 base_paddings = np.zeros((num_block_dims, 2), np.int32)
[all …]
/external/tensorflow/tensorflow/core/kernels/
Dspacetobatch_benchmark_test.cc27 const int num_block_dims = 2; in ConstructSpaceToBatchGraph() local
28 CHECK_EQ(num_block_dims, paddings.size()); in ConstructSpaceToBatchGraph()
30 Tensor paddings_tensor(DT_INT32, TensorShape({num_block_dims, 2})); in ConstructSpaceToBatchGraph()
32 for (int block_dim = 0; block_dim < num_block_dims; ++block_dim) { in ConstructSpaceToBatchGraph()
/external/tensorflow/tensorflow/core/ops/
Darray_ops.cc2090 const int64 num_block_dims = c->Value(num_block_dims_handle); in SpaceToBatchShapeHelper() local
2093 c->WithRankAtLeast(input_shape, num_block_dims + 1, &input_shape)); in SpaceToBatchShapeHelper()
2096 c->Merge(paddings_shape, c->Matrix(num_block_dims, 2), &paddings_shape)); in SpaceToBatchShapeHelper()
2102 for (int64 dim = 0; dim < num_block_dims; ++dim) { in SpaceToBatchShapeHelper()
2114 } else if (num_block_dims > 0) { in SpaceToBatchShapeHelper()
2119 output_dims.resize(num_block_dims + 1, c->UnknownDim()); in SpaceToBatchShapeHelper()
2123 for (int64 dim = 0; dim < num_block_dims; ++dim) { in SpaceToBatchShapeHelper()
2143 c->Subshape(input_shape, 1 + num_block_dims, &remaining_input_shape)); in SpaceToBatchShapeHelper()
2165 const int64 num_block_dims = c->Value(num_block_dims_handle); in BatchToSpaceShapeHelper() local
2168 c->WithRankAtLeast(input_shape, num_block_dims + 1, &input_shape)); in BatchToSpaceShapeHelper()
[all …]
/external/tensorflow/tensorflow/python/ops/
Darray_ops.py2644 num_block_dims = block_shape.get_shape().dims[0].value
2645 if num_block_dims == 0:
2648 input_shape.get_shape().assert_is_compatible_with([num_block_dims])
2653 base_paddings.get_shape().assert_is_compatible_with([num_block_dims, 2])
2655 base_paddings = zeros([num_block_dims, 2], dtypes.int32)
2674 [[pad_start[i], pad_end[i]] for i in range(num_block_dims)],
2677 [[0, pad_end_extra[i]] for i in range(num_block_dims)], name="crops")