Searched refs:block_shape_val (Results 1 – 2 of 2) sorted by relevance
909 int32_t block_shape_val = in convertSpaceToBatchNDOp() local914 a2_shape[1 + i * 2 + 0] = padded_shape[1 + i] / block_shape_val; in convertSpaceToBatchNDOp()915 a2_shape[1 + i * 2 + 1] = block_shape_val; in convertSpaceToBatchNDOp()916 block_num_elems *= block_shape_val; in convertSpaceToBatchNDOp()978 int32_t block_shape_val = in convertSpaceToBatchNDOp() local983 a4_reshape_shape[i + 1] = padded_shape[i + 1] / block_shape_val; in convertSpaceToBatchNDOp()1102 int block_shape_val = in convertBatchToSpaceNDOp() local1107 block_num_elems *= block_shape_val; in convertBatchToSpaceNDOp()1108 block_shape[i] = block_shape_val; in convertBatchToSpaceNDOp()
1303 const Value block_shape_val = operands[1]; in inferReturnTypes() local1306 const auto block_shape_type = block_shape_val.getType().cast<TensorType>(); in inferReturnTypes()1338 if (GetValueAsConstant(block_shape_val, block_shape_attr) && in inferReturnTypes()