Searched refs:space_to_depth (Results 1 – 17 of 17) sorted by relevance
/external/tensorflow/tensorflow/python/kernel_tests/ |
D | spacetodepth_op_test.py | 41 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 …]
|
D | depthtospace_op_test.py | 229 out_tf = array_ops.space_to_depth(x_np, block_size) 241 _ = array_ops.space_to_depth(x_np, block_size)
|
D | spacetobatch_op_test.py | 327 array_ops.space_to_depth(
|
/external/tensorflow/tensorflow/lite/kernels/ |
D | space_to_depth.cc | 26 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()
|
D | Android.bp | 102 "space_to_depth.cc",
|
D | BUILD | 214 "space_to_depth.cc",
|
/external/tensorflow/tensorflow/python/ops/ |
D | array_ops.py | 2709 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__
|
D | array_grad.py | 703 return array_ops.space_to_depth(grad, block_size, data_format=data_format)
|
/external/tensorflow/tensorflow/lite/ |
D | tflite_static.bp | 109 "kernels/space_to_depth.cc",
|
D | build_def.bzl | 313 "space_to_depth",
|
/external/tensorflow/tensorflow/lite/g3doc/guide/ |
D | ops_compatibility.md | 73 * [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/ |
D | tensorflow.nn.pbtxt | 296 name: "space_to_depth"
|
/external/tensorflow/tensorflow/tools/api/golden/v1/ |
D | tensorflow.nn.pbtxt | 376 name: "space_to_depth"
|
D | tensorflow.pbtxt | 2060 name: "space_to_depth"
|
/external/tensorflow/tensorflow/compiler/tests/ |
D | unary_ops_test.py | 973 return array_ops.space_to_depth(
|
/external/tensorflow/tensorflow/lite/testing/ |
D | generate_examples.py | 2522 out = tf.space_to_depth(input_tensor, block_size=parameters["block_size"])
|
/external/tensorflow/ |
D | RELEASE.md | 815 * Add `NCHW_VECT_C` support for `tf.space_to_depth` on GPU.
|