Searched refs:depth_to_space (Results 1 – 10 of 10) sorted by relevance
/external/tensorflow/tensorflow/python/kernel_tests/ |
D | depthtospace_op_test.py | 42 x_tf = array_ops.depth_to_space(input_nhwc, block_size) 47 x_tf = array_ops.depth_to_space(input_nhwc, block_size) 51 output_nchw = array_ops.depth_to_space( 111 x_tf = array_ops.depth_to_space(input_nhwc, block_size) 117 x_tf = array_ops.depth_to_space(input_nhwc, block_size) 197 out_tf = array_ops.depth_to_space(x_np, block_size) 207 out_tf = array_ops.depth_to_space(x_np, block_size) 219 out_tf = array_ops.depth_to_space(x_np, block_size) 245 t = array_ops.depth_to_space( 283 t = array_ops.depth_to_space(t, block_size, data_format="NCHW_VECT_C") [all …]
|
D | batchtospace_op_test.py | 62 array_ops.depth_to_space(
|
/external/tensorflow/tensorflow/python/ops/ |
D | array_ops.py | 2726 def depth_to_space(input, block_size, name=None, data_format="NHWC"): # pylint: disable=redefined-… function 2727 return gen_array_ops.depth_to_space(input, block_size, data_format, name=name) 2730 depth_to_space.__doc__ = gen_array_ops.depth_to_space.__doc__ 2735 return gen_array_ops.depth_to_space(input, block_size, data_format, name=name) 2738 depth_to_space_v2.__doc__ = gen_array_ops.depth_to_space.__doc__
|
D | array_grad.py | 692 return array_ops.depth_to_space(grad, block_size, data_format=data_format)
|
/external/tensorflow/tensorflow/lite/g3doc/guide/ |
D | ops_compatibility.md | 136 * [tf.depth_to_space](https://www.tensorflow.org/api_docs/python/tf/depth_to_space)
|
/external/tensorflow/tensorflow/tools/api/golden/v2/ |
D | tensorflow.nn.pbtxt | 120 name: "depth_to_space"
|
/external/tensorflow/tensorflow/tools/api/golden/v1/ |
D | tensorflow.nn.pbtxt | 144 name: "depth_to_space"
|
D | tensorflow.pbtxt | 1072 name: "depth_to_space"
|
/external/tensorflow/tensorflow/compiler/tests/ |
D | unary_ops_test.py | 921 return array_ops.depth_to_space(
|
/external/tensorflow/ |
D | RELEASE.md | 814 * Add `NCHW_VECT_C` support for `tf.depth_to_space` on GPU. 901 * Add NCHW support to `tf.depth_to_space` on GPU.
|