/external/tensorflow/tensorflow/cc/gradients/ |
array_grad_test.cc | 287 auto block_shape = Const(scope_, {2, 2}); local 290 auto y = SpaceToBatchND(scope_, x, block_shape, paddings); 306 auto block_shape = Const(scope_, {2, 2}); local 309 auto y = BatchToSpaceND(scope_, x, block_shape, paddings);
|
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
batchtospace_op.cc | 26 absl::Span<const int64> block_shape, 31 const int block_rank = block_shape.size(); 51 // Compute the product of the block_shape values. 54 block_num_elems *= block_shape[i]; 61 // [block_shape[0], ..., block_shape[M-1], 62 // batch / prod(block_shape), 71 std::copy(block_shape.begin(), block_shape.end(), reshaped_shape.begin()); 78 // [batch / prod(block_shape), 151 std::vector<int64> block_shape; variable [all...] |
spacetobatch_op.cc | 26 absl::Span<const int64> block_shape, 31 const int block_rank = block_shape.size(); 66 block_num_elems *= block_shape[i]; 82 // [padded_shape[1] / block_shape[0], 83 // block_shape[0], 85 // padded_shape[M] / block_shape[M-1], 86 // block_shape[M-1]] + 92 OP_REQUIRES(ctx, padded_shape[1 + i] % block_shape[i] == 0, 95 " is not divisible by block_shape[", i, 96 "]=", block_shape[i])) 153 std::vector<int64> block_shape; variable [all...] |
/external/tensorflow/tensorflow/core/kernels/ |
spacetobatch_functor.cc | 40 const int64* space_tensor_strides, const int64* block_shape, 47 batch_tensor_pos * block_shape[0] + block_offsets[0] - pad_start[0]; 51 space_tensor_shape + 1, space_tensor_strides + 1, block_shape + 1, 71 const int64* space_tensor_strides, const int64* block_shape, 104 int64 block_shape[NUM_BLOCK_DIMS]; local 109 block_shape[block_dim] = block_shape_tensor[block_dim]; 138 block_dim > 0 ? block_index % block_shape[block_dim] : block_index; 139 block_index /= block_shape[block_dim]; 145 space_tensor_shape, &space_tensor_strides[1], block_shape, pad_start,
|
spacetobatch_functor_gpu.cu.cc | 39 int32 block_shape[NUM_BLOCK_DIMS]; member in struct:tensorflow::S2BParameters 70 offset %= args.block_shape[block_dim]; 73 batch_tensor_pos[block_dim + 1] * args.block_shape[block_dim] + 95 remaining_block_idx /= args.block_shape[block_dim]; 108 const int64 block_shape[NUM_BLOCK_DIMS], 118 if (block_shape[block_dim] > std::numeric_limits<int32>::max()) { 119 return errors::InvalidArgument("block_shape value exceeds 2^32-1"); 121 args.block_shape[block_dim] = block_shape[block_dim];
|
batchtospace_op.cc | 50 errors::InvalidArgument("block_shape rank should be 1 instead of ", 66 // To avoid out-of-bounds access in the case that the block_shape and/or 68 gtl::InlinedVector<int64, 4> block_shape; local 70 internal::spacetobatch::SubtleMustCopyFlat(orig_block_shape, &block_shape); 74 // into the batch dimension due to having no padding and block_shape=1. 79 block_shape[dim] != 1) { 85 // into the depth dimension due to having no padding and block_shape=1. 91 block_shape[dim] != 1) { 96 // Compute the product of the block_shape values. 99 block_shape_product *= block_shape[block_dim] [all...] |
spacetobatch_op.cc | 51 return errors::InvalidArgument("block_shape rank should be 1 instead of ", 69 // To avoid out-of-bounds access in the case that the block_shape and/or 71 gtl::InlinedVector<int64, 4> block_shape; local 73 internal::spacetobatch::SubtleMustCopyFlat(orig_block_shape, &block_shape); 77 // into the batch dimension due to having no padding and block_shape=1. 82 block_shape[dim] != 1) { 88 // into the depth dimension due to having no padding and block_shape=1. 94 block_shape[dim] != 1) { 99 // Compute the product of the block_shape values. 102 block_shape_product *= block_shape[block_dim] [all...] |
/external/tensorflow/tensorflow/core/ops/ |
array_ops_test.cc | 1378 Tensor block_shape = test::AsTensor<int32>({2, 3}); local 1405 Tensor block_shape = test::AsTensor<int32>({0, 2}); local 1412 Tensor block_shape = test::AsTensor<int32>({1, 1}); local 1422 Tensor block_shape = test::AsTensor<int32>({3, 3}); local 1495 Tensor block_shape = test::AsTensor<int32>({2, 3}); local 1526 Tensor block_shape = test::AsTensor<int32>({0, 2}); local 1533 Tensor block_shape = test::AsTensor<int32>({1, 1}); local 1544 Tensor block_shape = test::AsTensor<int32>({2, 2}); local 1555 Tensor block_shape = test::AsTensor<int32>({2, 3}); local [all...] |
/external/tensorflow/tensorflow/lite/kernels/ |
batch_to_space_nd.cc | 39 block_shape = GetInput(context, node, 1); 44 const TfLiteTensor* block_shape; member in struct:tflite::ops::builtin::batch_to_space_nd::BatchToSpaceNDContext 59 const int* block_shape = GetTensorData<int32>(op_context->block_shape); local 62 TF_LITE_ENSURE_EQ(context, NumDimensions(op_context->block_shape), 64 TF_LITE_ENSURE_EQ(context, op_context->block_shape->dims->data[0], 74 // Number of batch must be multiple of (block_shape[0] * block_shape[1]). 76 input_size->data[0] % (block_shape[0] * block_shape[1]), 0) [all...] |
space_to_batch_nd.cc | 39 block_shape = GetInput(context, node, 1); 44 const TfLiteTensor* block_shape; member in struct:tflite::ops::builtin::space_to_batch_nd::SpaceToBatchNDContext 59 const int32* block_shape = GetTensorData<int32>(op_context->block_shape); local 62 TF_LITE_ENSURE_EQ(context, NumDimensions(op_context->block_shape), 64 TF_LITE_ENSURE_EQ(context, op_context->block_shape->dims->data[0], 76 TF_LITE_ENSURE_EQ(context, final_dim_size % block_shape[dim], 0); 77 output_size->data[dim + 1] = final_dim_size / block_shape[dim]; 81 input_size->data[0] * block_shape[0] * block_shape[1] [all...] |
/external/tensorflow/tensorflow/python/ops/linalg/ |
linear_operator_circulant.py | 122 # block_shape = [s1, s2] 193 def block_shape(self): member in class:_BaseLinearOperatorCirculant 206 # self.block_shape = [b0, b1] (note b0 * b1 = m2). 219 self.block_shape.is_fully_defined()): 223 final_shape = vec_leading_shape.concatenate(self.block_shape) 384 # spectrum.shape = batch_shape + block_shape 386 # spectrum.shape = batch_shape + [prod(block_shape), 1] [all...] |
/external/tensorflow/tensorflow/lite/toco/ |
model.h | 1815 std::vector<int> block_shape; member in struct:toco::SpaceToBatchNDOperator 1832 std::vector<int> block_shape; member in struct:toco::BatchToSpaceNDOperator [all...] |