Home
last modified time | relevance | path

Searched refs:space_to_depth (Results 1 – 17 of 17) sorted by relevance

/external/tensorflow/tensorflow/python/kernel_tests/
Dspacetodepth_op_test.py41 x_tf = array_ops.space_to_depth(input_nhwc, block_size)
47 x_tf = array_ops.space_to_depth(input_nhwc, block_size)
51 output_nchw = array_ops.space_to_depth(
140 x_tf = array_ops.space_to_depth(input_nhwc, block_size)
147 x_tf = array_ops.space_to_depth(input_nhwc, block_size)
168 out_tf = array_ops.space_to_depth(x_np, block_size)
177 _ = array_ops.space_to_depth(x_np, block_size)
185 out_tf = array_ops.space_to_depth(x_np, block_size)
194 out_tf = array_ops.space_to_depth(x_np, block_size)
203 out_tf = array_ops.space_to_depth(x_np, block_size)
[all …]
Ddepthtospace_op_test.py229 out_tf = array_ops.space_to_depth(x_np, block_size)
241 _ = array_ops.space_to_depth(x_np, block_size)
Dspacetobatch_op_test.py327 array_ops.space_to_depth(
/external/tensorflow/tensorflow/lite/kernels/
Dspace_to_depth.cc26 namespace space_to_depth { namespace
139 nullptr, nullptr, space_to_depth::Prepare, in Register_SPACE_TO_DEPTH_REF()
140 space_to_depth::Eval<space_to_depth::kReference>}; in Register_SPACE_TO_DEPTH_REF()
146 nullptr, nullptr, space_to_depth::Prepare, in Register_SPACE_TO_DEPTH_GENERIC_OPT()
147 space_to_depth::Eval<space_to_depth::kGenericOptimized>}; in Register_SPACE_TO_DEPTH_GENERIC_OPT()
DAndroid.bp102 "space_to_depth.cc",
DBUILD214 "space_to_depth.cc",
/external/tensorflow/tensorflow/python/ops/
Darray_ops.py2709 def space_to_depth(input, block_size, name=None, data_format="NHWC"): # pylint: disable=redefined-… function
2710 return gen_array_ops.space_to_depth(input, block_size, data_format, name=name)
2713 space_to_depth.__doc__ = gen_array_ops.space_to_depth.__doc__
2718 return gen_array_ops.space_to_depth(input, block_size, data_format, name=name)
2721 space_to_depth_v2.__doc__ = gen_array_ops.space_to_depth.__doc__
Darray_grad.py703 return array_ops.space_to_depth(grad, block_size, data_format=data_format)
/external/tensorflow/tensorflow/lite/
Dtflite_static.bp109 "kernels/space_to_depth.cc",
Dbuild_def.bzl313 "space_to_depth",
/external/tensorflow/tensorflow/lite/g3doc/guide/
Dops_compatibility.md73 * [tf.space_to_depth](https://www.tensorflow.org/api_docs/python/tf/space_to_depth)
882 0: a tensor rearranged using block_size. See tf.space_to_depth for details.
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.nn.pbtxt296 name: "space_to_depth"
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.nn.pbtxt376 name: "space_to_depth"
Dtensorflow.pbtxt2060 name: "space_to_depth"
/external/tensorflow/tensorflow/compiler/tests/
Dunary_ops_test.py973 return array_ops.space_to_depth(
/external/tensorflow/tensorflow/lite/testing/
Dgenerate_examples.py2522 out = tf.space_to_depth(input_tensor, block_size=parameters["block_size"])
/external/tensorflow/
DRELEASE.md815 * Add `NCHW_VECT_C` support for `tf.space_to_depth` on GPU.