Lines Matching refs:block_shape
1399 "block_shape": [[1, 3], [2, 2]],
1407 "block_shape": [[1, 3], [2, 2]],
1416 "block_shape": [[2, 2, 2]],
1431 # Get block_shape either as a const or as a placeholder (tensor).
1433 block_shape = parameters["block_shape"]
1435 shape = [len(parameters["block_shape"])]
1436 block_shape = tf.placeholder(dtype=tf.int32, name="shape", shape=shape)
1437 input_tensors.append(block_shape)
1447 out = tf.space_to_batch_nd(input_tensor, block_shape, paddings)
1455 values.append(np.array(parameters["block_shape"]))
1470 "block_shape": [[1, 4], [2, 2], [3, 4]],
1479 "block_shape": [[2, 2, 2]],
1494 # Get block_shape either as a const or as a placeholder (tensor).
1496 block_shape = parameters["block_shape"]
1498 shape = [len(parameters["block_shape"])]
1499 block_shape = tf.placeholder(dtype=tf.int32, name="shape", shape=shape)
1500 input_tensors.append(block_shape)
1510 out = tf.batch_to_space_nd(input_tensor, block_shape, crops)
1518 values.append(np.array(parameters["block_shape"]))
