Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/python/kernel_tests/
Dspacetodepth_op_test.py42 x_tf = array_ops.space_to_depth(input_nhwc, block_size)
48 x_tf = array_ops.space_to_depth(input_nhwc, block_size)
52 output_nchw = array_ops.space_to_depth(
141 x_tf = array_ops.space_to_depth(input_nhwc, block_size)
148 x_tf = array_ops.space_to_depth(input_nhwc, block_size)
168 out_tf = array_ops.space_to_depth(x_np, block_size)
176 _ = array_ops.space_to_depth(x_np, block_size)
183 out_tf = array_ops.space_to_depth(x_np, block_size)
191 out_tf = array_ops.space_to_depth(x_np, block_size)
199 out_tf = array_ops.space_to_depth(x_np, block_size)
[all …]
Ddepthtospace_op_test.py243 out_tf = array_ops.space_to_depth(x_np, block_size)
255 _ = array_ops.space_to_depth(x_np, block_size)
Dspacetobatch_op_test.py327 array_ops.space_to_depth(
/external/tensorflow/tensorflow/lite/micro/kernels/
Dspace_to_depth.cc29 namespace space_to_depth { namespace
148 nullptr, nullptr, space_to_depth::Prepare, in Register_SPACE_TO_DEPTH_REF()
149 space_to_depth::Eval<space_to_depth::kReference>}; in Register_SPACE_TO_DEPTH_REF()
155 nullptr, nullptr, space_to_depth::Prepare, in Register_SPACE_TO_DEPTH_GENERIC_OPT()
156 space_to_depth::Eval<space_to_depth::kGenericOptimized>}; in Register_SPACE_TO_DEPTH_GENERIC_OPT()
/external/tensorflow/tensorflow/lite/kernels/
Dspace_to_depth.cc29 namespace space_to_depth { namespace
148 nullptr, nullptr, space_to_depth::Prepare, in Register_SPACE_TO_DEPTH_REF()
149 space_to_depth::Eval<space_to_depth::kReference>}; in Register_SPACE_TO_DEPTH_REF()
155 nullptr, nullptr, space_to_depth::Prepare, in Register_SPACE_TO_DEPTH_GENERIC_OPT()
156 space_to_depth::Eval<space_to_depth::kGenericOptimized>}; in Register_SPACE_TO_DEPTH_GENERIC_OPT()
DAndroid.bp148 "space_to_depth.cc",
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dtpu_space_to_depth_pass.cc443 auto space_to_depth = in HandleHostInput() local
445 cluster_func.setOperand(index, space_to_depth); in HandleHostInput()
446 return space_to_depth; in HandleHostInput()
470 auto space_to_depth = in HandleHostReplicatedInputs() local
472 entry.value().set(space_to_depth); in HandleHostReplicatedInputs()
473 block_arg.setType(space_to_depth.getType()); in HandleHostReplicatedInputs()
/external/tensorflow/tensorflow/lite/testing/op_tests/
Dspace_to_depth.py47 out = tf.space_to_depth(input_tensor, block_size=parameters["block_size"])
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/g3doc/
Dspace_to_depth.md56 `tf.nn.space_to_depth`.
102 … follows the space-to-depth logic: https://www.tensorflow.org/api_docs/python/tf/nn/space_to_depth)
162 %space_to_depth = "tf.SpaceToDepth"(%input) {block_size = 2, ...}:
164 %device_launch = "tf_device.launch_func"(%space_to_depth,...) {func = @_func,...)
/external/tensorflow/tensorflow/lite/delegates/gpu/gl/kernels/
DBUILD628 name = "space_to_depth",
629 srcs = ["space_to_depth.cc"],
630 hdrs = ["space_to_depth.h"],
649 ":space_to_depth",
767 "space_to_depth",
/external/tensorflow/tensorflow/python/ops/
Darray_ops.py3985 def space_to_depth(input, block_size, name=None, data_format="NHWC"): # pylint: disable=redefined-… function
3986 return gen_array_ops.space_to_depth(input, block_size, data_format, name=name)
3989 space_to_depth.__doc__ = gen_array_ops.space_to_depth.__doc__
3995 return gen_array_ops.space_to_depth(input, block_size, data_format, name=name)
3998 space_to_depth_v2.__doc__ = gen_array_ops.space_to_depth.__doc__
Darray_grad.py958 return array_ops.space_to_depth(grad, block_size, data_format=data_format)
/external/tensorflow/tensorflow/lite/
Dtflite_static.bp131 "kernels/space_to_depth.cc",
Dbuild_def.bzl390 "space_to_depth",
/external/tensorflow/tensorflow/lite/delegates/gpu/common/selectors/
DBUILD138 "//tensorflow/lite/delegates/gpu/common/tasks:space_to_depth",
/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/
DBUILD839 name = "space_to_depth",
840 srcs = ["space_to_depth.cc"],
841 hdrs = ["space_to_depth.h"],
857 ":space_to_depth",
/external/tensorflow/tensorflow/lite/testing/
Dgenerate_examples_lib.py134 from tensorflow.lite.testing.op_tests.space_to_depth import make_space_to_depth_tests
DBUILD638 "space_to_depth",
/external/tensorflow/tensorflow/lite/kernels/internal/
DBUILD507 "reference/space_to_depth.h",
616 "reference/space_to_depth.h",
/external/tensorflow/tensorflow/python/ops/parallel_for/
Darray_test.py490 return array_ops.space_to_depth(x1, 2, data_format="NHWC")
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.nn.pbtxt312 name: "space_to_depth"
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.nn.pbtxt384 name: "space_to_depth"
/external/tensorflow/tensorflow/compiler/tests/
Dunary_ops_test.py1120 return array_ops.space_to_depth(
/external/tensorflow/tensorflow/compiler/mlir/lite/tests/
Dlegalize-tf.mlir1404 func @space_to_depth(%arg0: tensor<1x2x2x1xf32>) -> tensor<?xf32> {
1408 // CHECK-LABEL: space_to_depth
1410 …// CHECK: "tfl.space_to_depth"(%[[ARG]]) {block_size = 2 : i32} : (tensor<1x2x2x1xf32>) -> tensor<…
/external/tensorflow/tensorflow/compiler/mlir/tosa/tests/
Dtfl-to-tosa-pipeline.mlir727 …%0 = "tfl.space_to_depth"(%arg0) {block_size = 2 : i32} : (tensor<1x32x32x8xf32>) -> tensor<1x16…

12