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.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.cc1231 const int num_block_dims = 2; in TEST_F() local
1233 RandomDims(num_block_dims, num_block_dims, 0, 5); in TEST_F()
1236 std::vector<int64> input_dims(1 + num_block_dims + 1); in TEST_F()
1238 for (int i = 0; i < num_block_dims; ++i) { in TEST_F()
1242 input_dims[1 + num_block_dims] = RandomDim(); in TEST_F()
1246 for (int i = 0; i < num_block_dims; ++i) { in TEST_F()
1253 TensorShape({num_block_dims, 2}))); in TEST_F()
1267 int num_block_dims = block_dims.size(); in TEST_F() local
1272 std::vector<int64> input_dims(1 + num_block_dims + remaining_dims.size()); in TEST_F()
1274 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/lite/tools/optimize/sparsity/
Dformat_converter.cc85 int num_block_dims = block_map_.size(); in DenseToSparse() local
86 int num_expanded_dims = num_original_dims + num_block_dims; in DenseToSparse()
106 for (int i = 0; i < num_block_dims; ++i) { in DenseToSparse()
/external/tensorflow/tensorflow/core/ops/
Darray_ops.cc2247 const int64 num_block_dims = c->Value(num_block_dims_handle); in SpaceToBatchShapeHelper() local
2250 c->WithRankAtLeast(input_shape, num_block_dims + 1, &input_shape)); in SpaceToBatchShapeHelper()
2253 c->Merge(paddings_shape, c->Matrix(num_block_dims, 2), &paddings_shape)); in SpaceToBatchShapeHelper()
2259 for (int64 dim = 0; dim < num_block_dims; ++dim) { in SpaceToBatchShapeHelper()
2271 } else if (num_block_dims > 0) { in SpaceToBatchShapeHelper()
2276 output_dims.resize(num_block_dims + 1, c->UnknownDim()); in SpaceToBatchShapeHelper()
2280 for (int64 dim = 0; dim < num_block_dims; ++dim) { in SpaceToBatchShapeHelper()
2300 c->Subshape(input_shape, 1 + num_block_dims, &remaining_input_shape)); in SpaceToBatchShapeHelper()
2322 const int64 num_block_dims = c->Value(num_block_dims_handle); in BatchToSpaceShapeHelper() local
2325 c->WithRankAtLeast(input_shape, num_block_dims + 1, &input_shape)); in BatchToSpaceShapeHelper()
[all …]
/external/tensorflow/tensorflow/python/ops/
Darray_ops.py3912 num_block_dims = block_shape.get_shape().dims[0].value
3913 if num_block_dims == 0:
3916 input_shape.get_shape().assert_is_compatible_with([num_block_dims])
3921 base_paddings.get_shape().assert_is_compatible_with([num_block_dims, 2])
3923 base_paddings = zeros([num_block_dims, 2], dtypes.int32)
3942 [[pad_start[i], pad_end[i]] for i in range(num_block_dims)],
3944 result_crops = stack([[0, pad_end_extra[i]] for i in range(num_block_dims)],