Searched refs:block_shape_value (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | spacetobatch_op.cc | 154 const int64 block_shape_value = block_shape[block_dim]; in SpaceToBatchOpCompute() local
156 if (padded_size % block_shape_value != 0) { in SpaceToBatchOpCompute()
160 block_dim, "]=", block_shape_value); in SpaceToBatchOpCompute()
163 const int64 output_size = padded_size / block_shape_value; in SpaceToBatchOpCompute()
|
| D | batchtospace_op.cc | 155 const int64 block_shape_value = block_shape[block_dim]; in BatchToSpaceOpCompute() local
157 input_size * block_shape_value - crop_start - crop_end; in BatchToSpaceOpCompute()
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | spacetobatch_op_test.py | 54 for block_dim, block_shape_value in enumerate(block_shape):
55 reduced_size = padded.shape[block_dim + 1] // block_shape_value
58 reshaped_padded_shape.append(block_shape_value)
|
| /external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/ |
| D | legalize_common.h | 82 Value block_shape_value,
89 Value block_shape_value,
|
| D | legalize_common.cc | 734 Value block_shape_value, in convertSpaceToBatchNDOp() argument
794 block_shape_value.getType().dyn_cast<RankedTensorType>(); in convertSpaceToBatchNDOp()
836 if (!matchPattern(block_shape_value, m_Constant(&block_shape_elems))) in convertSpaceToBatchNDOp()
1002 Value block_shape_value, in convertBatchToSpaceNDOp() argument
1055 block_shape_value.getType().dyn_cast<RankedTensorType>(); in convertBatchToSpaceNDOp()
1086 if (!matchPattern(block_shape_value, m_Constant(&block_shape_elems))) { in convertBatchToSpaceNDOp()
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | spacetobatch_op_test.py | 58 for block_dim, block_shape_value in enumerate(block_shape):
59 reduced_size = padded.shape[block_dim + 1] // block_shape_value
62 reshaped_padded_shape.append(block_shape_value)
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | array_ops.cc | 2260 const int64 block_shape_value = block_shape_vec[dim]; in SpaceToBatchShapeHelper() local
2261 if (block_shape_value < 1) { in SpaceToBatchShapeHelper()
2266 c->Multiply(batch_size, block_shape_value, &batch_size)); in SpaceToBatchShapeHelper()
2335 const int64 block_shape_value = block_shape_vec[dim]; in BatchToSpaceShapeHelper() local
2336 if (block_shape_value < 1) { in BatchToSpaceShapeHelper()
2340 TF_RETURN_IF_ERROR(c->Divide(batch_size, block_shape_value, in BatchToSpaceShapeHelper()
|