| /external/tensorflow/tensorflow/python/kernel_tests/ |
| batchtospace_op_test.py | 40 def batch_to_space(*args, **kwargs): member in class:PythonOpImpl
41 return array_ops.batch_to_space(*args, **kwargs)
47 def batch_to_space(*args, **kwargs): member in class:CppOpImpl
48 return gen_array_ops.batch_to_space(*args, **kwargs)
53 # Verifies that: batch_to_space(x) = transpose(depth_to_space(transpose(x)))
60 y1 = self.batch_to_space(x, crops, block_size=block_size)
82 _ = self.batch_to_space(x_np, crops, block_size)
91 out_tf = self.batch_to_space(x_np, crops, block_size)
101 out_tf = self.batch_to_space(x_np, crops, block_size)
111 out_tf = self.batch_to_space(x_np, crops, block_size
[all...] |
| spacetobatch_op_test.py | 82 def batch_to_space(*args, **kwargs): member in class:PythonOpImpl
83 return array_ops.batch_to_space(*args, **kwargs)
93 def batch_to_space(*args, **kwargs): member in class:CppOpImpl
94 return gen_array_ops.batch_to_space(*args, **kwargs)
111 # inputs = batch_to_space(outputs)
112 x_tf = self.batch_to_space(
209 # inputs = batch_to_space(outputs)
|
| atrous_conv2d_test.py | 107 net = batch_to_space(net, crops=pad, block_size=rate)
129 # y2: space_to_batch, three conv2d in a row, batch_to_space
137 y2 = array_ops.batch_to_space(y2, crops=pad, block_size=rate)
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| spacetobatch_op_test.py | 82 # inputs = batch_to_space(outputs)
83 x_tf = gen_array_ops.batch_to_space(
180 # inputs = batch_to_space(outputs)
|
| /external/tensorflow/tensorflow/contrib/gan/python/eval/python/ |
| sliced_wasserstein_impl.py | 74 res = array_ops.batch_to_space(res, crops=[[0, 0], [0, 0]], block_size=2)
|
| /external/tensorflow/tensorflow/python/ops/ |
| array_ops.py | 2742 def batch_to_space(input, crops, block_size, name=None, block_shape=None): # pylint: disable=redefined-builtin,missing-docstring function
[all...] |
| array_grad.py | 655 array_ops.batch_to_space(grad, op.inputs[1], block_size=block_size), None
[all...] |
| nn_ops.py | [all...] |
