__init__.py « space_to_batch_nd « examples « TensorFlowPythonExamples « res - platform/core/ml/nnfw - Domain: Machine Learning / ML Framework; Licenses: Apache-2.0;

index : platform/core/ml/nnfw

Domain: Machine Learning / ML Framework; Licenses: Apache-2.0;

Sangmin Seo <sangmin7.seo@samsung.com>, Sung-Jae Lee <sjlee@mail.com>

summary refs log tree commit diff

path: root/res/TensorFlowPythonExamples/examples/space_to_batch_nd/__init__.py

blob: 892796b122150f52b581c4ad27369cd6403a3ef4 (plain)

import tensorflow as tf

tf.compat.v1.disable_eager_execution()

in_ = tf.compat.v1.placeholder(tf.float32, shape=[1, 2, 2, 1], name="Hole")
bs_ = tf.constant([2, 2], name="Hole")
pd_ = tf.constant([[0, 0], [0, 0]], name="Hole")
op_ = tf.space_to_batch_nd(in_, bs_, pd_)