Home
last modified time | relevance | path

Searched refs:depth_to_space (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/python/kernel_tests/
Ddepthtospace_op_test.py42 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 …]
Dbatchtospace_op_test.py62 array_ops.depth_to_space(
/external/tensorflow/tensorflow/python/ops/
Darray_ops.py2726 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__
Darray_grad.py692 return array_ops.depth_to_space(grad, block_size, data_format=data_format)
/external/tensorflow/tensorflow/lite/g3doc/guide/
Dops_compatibility.md136 * [tf.depth_to_space](https://www.tensorflow.org/api_docs/python/tf/depth_to_space)
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.nn.pbtxt120 name: "depth_to_space"
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.nn.pbtxt144 name: "depth_to_space"
Dtensorflow.pbtxt1072 name: "depth_to_space"
/external/tensorflow/tensorflow/compiler/tests/
Dunary_ops_test.py921 return array_ops.depth_to_space(
/external/tensorflow/
DRELEASE.md814 * Add `NCHW_VECT_C` support for `tf.depth_to_space` on GPU.
901 * Add NCHW support to `tf.depth_to_space` on GPU.