Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/tests/
Dspacetobatch_op_test.py42 num_block_dims = len(block_shape)
47 (input_array.ndim - 1 - num_block_dims)),
56 reshaped_padded_shape.extend(input_array.shape[num_block_dims + 1:])
57 output_shape.extend(input_array.shape[num_block_dims + 1:])
61 list(np.arange(num_block_dims) * 2 + 2) + [0] +
62 list(np.arange(num_block_dims) * 2 + 1) + list(
63 np.arange(input_array.ndim - num_block_dims - 1) + 1 + num_block_dims
Drandomized_tests.cc1887 const int num_block_dims = 2; in TEST_F() local
1889 RandomDims(num_block_dims, num_block_dims, 0, 5); in TEST_F()
1892 std::vector<int64_t> input_dims(1 + num_block_dims + 1); in TEST_F()
1894 for (int i = 0; i < num_block_dims; ++i) { in TEST_F()
1898 input_dims[1 + num_block_dims] = RandomDim(); in TEST_F()
1902 for (int i = 0; i < num_block_dims; ++i) { in TEST_F()
1909 TensorShape({num_block_dims, 2}))); in TEST_F()
1923 int num_block_dims = block_dims.size(); in TEST_F() local
1928 std::vector<int64_t> input_dims(1 + num_block_dims + remaining_dims.size()); in TEST_F()
1930 for (int i = 0; i < num_block_dims; ++i) { in TEST_F()
[all …]
/external/tensorflow/tensorflow/python/kernel_tests/array_ops/
Dspacetobatch_op_test.py47 num_block_dims = len(block_shape)
52 (input_array.ndim - 1 - num_block_dims)),
61 reshaped_padded_shape.extend(input_array.shape[num_block_dims + 1:])
62 output_shape.extend(input_array.shape[num_block_dims + 1:])
66 list(np.arange(num_block_dims) * 2 + 2) + [0] +
67 list(np.arange(num_block_dims) * 2 + 1) + list(
68 np.arange(input_array.ndim - num_block_dims - 1) + 1 + num_block_dims
651 num_block_dims = len(block_shape)
652 self.assertEqual(len(input_shape), num_block_dims)
654 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/lite/kernels/internal/utils/
Dsparsity_format_converter.cc87 int num_block_dims = block_map_.size(); in DenseToSparse() local
88 int num_expanded_dims = num_original_dims + num_block_dims; in DenseToSparse()
108 for (int i = 0; i < num_block_dims; ++i) { in DenseToSparse()
/external/tensorflow/tensorflow/core/ops/
Darray_ops.cc2280 const int64_t num_block_dims = c->Value(num_block_dims_handle); in SpaceToBatchShapeHelper() local
2283 c->WithRankAtLeast(input_shape, num_block_dims + 1, &input_shape)); in SpaceToBatchShapeHelper()
2286 c->Merge(paddings_shape, c->Matrix(num_block_dims, 2), &paddings_shape)); in SpaceToBatchShapeHelper()
2292 for (int64_t dim = 0; dim < num_block_dims; ++dim) { in SpaceToBatchShapeHelper()
2304 } else if (num_block_dims > 0) { in SpaceToBatchShapeHelper()
2309 output_dims.resize(num_block_dims + 1, c->UnknownDim()); in SpaceToBatchShapeHelper()
2313 for (int64_t dim = 0; dim < num_block_dims; ++dim) { in SpaceToBatchShapeHelper()
2333 c->Subshape(input_shape, 1 + num_block_dims, &remaining_input_shape)); in SpaceToBatchShapeHelper()
2355 const int64_t num_block_dims = c->Value(num_block_dims_handle); in BatchToSpaceShapeHelper() local
2358 c->WithRankAtLeast(input_shape, num_block_dims + 1, &input_shape)); in BatchToSpaceShapeHelper()
[all …]
/external/tensorflow/tensorflow/python/ops/
Darray_ops.py4038 num_block_dims = block_shape.get_shape().dims[0].value
4039 if num_block_dims == 0:
4042 input_shape.get_shape().assert_is_compatible_with([num_block_dims])
4047 base_paddings.get_shape().assert_is_compatible_with([num_block_dims, 2])
4049 base_paddings = zeros([num_block_dims, 2], dtypes.int32)
4068 [[pad_start[i], pad_end[i]] for i in range(num_block_dims)],
4070 result_crops = stack([[0, pad_end_extra[i]] for i in range(num_block_dims)],