Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/
Dlegalize_common.cc848 int32_t block_shape_val = in convertSpaceToBatchNDOp() local
852 a2_shape[1 + i * 2 + 0] /= block_shape_val; in convertSpaceToBatchNDOp()
855 a2_shape[1 + i * 2 + 1] = block_shape_val; in convertSpaceToBatchNDOp()
856 block_num_elems *= block_shape_val; in convertSpaceToBatchNDOp()
918 int32_t block_shape_val = in convertSpaceToBatchNDOp() local
922 a4_reshape_shape[i + 1] /= block_shape_val; in convertSpaceToBatchNDOp()
1042 int block_shape_val = in convertBatchToSpaceNDOp() local
1047 block_num_elems *= block_shape_val; in convertBatchToSpaceNDOp()
1048 block_shape[i] = block_shape_val; in convertBatchToSpaceNDOp()