| /external/tensorflow/tensorflow/core/api_def/java_api/ |
| D | api_def_SpaceToDepth.pbtxt | 2 graph_op_name: "SpaceToDepth"
4 name: "nn.SpaceToDepth"
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/gl/kernels/ |
| D | space_to_depth.cc | 32 class SpaceToDepth : public NodeShader { class
68 return absl::make_unique<SpaceToDepth>(); in NewSpaceToDepthNodeShader()
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | SpaceToDepth.pbtxt | 2 name: "SpaceToDepth"
23 name: "SpaceToDepth"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | SpaceToDepth.pbtxt | 2 name: "SpaceToDepth"
23 name: "SpaceToDepth"
|
| /external/tensorflow/tensorflow/core/api_def/python_api/ |
| D | api_def_SpaceToDepth.pbtxt | 2 graph_op_name: "SpaceToDepth"
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_SpaceToDepth.pbtxt | 2 graph_op_name: "SpaceToDepth"
9 summary: "SpaceToDepth for tensors of type T."
|
| D | api_def_DepthToSpace.pbtxt | 12 This is the reverse transformation of SpaceToDepth. More specifically,
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/g3doc/ |
| D | space_to_depth.md | 162 %space_to_depth = "tf.SpaceToDepth"(%input) {block_size = 2, ...}:
178 ### SpaceToDepth Trigger Condition
186 ### Fuse SpaceToDepth with Automatic Double Transpose
194 SpaceToDepth not only helps with 2D image models but also 3D image models such
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | space_to_depth.h | 24 inline void SpaceToDepth(const tflite::SpaceToDepthParams& op_params, in SpaceToDepth() function
|
| D | legacy_reference_ops.h | 1856 inline void SpaceToDepth(const T* input_data, const Dims<4>& input_dims, in SpaceToDepth() function
1862 SpaceToDepth(op_params, DimsToShape(input_dims), input_data, in SpaceToDepth()
|
| /external/tensorflow/tensorflow/lite/micro/kernels/ |
| D | space_to_depth.cc | 95 type::SpaceToDepth(op_params, GetTensorShape(input), \ in Eval()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | space_to_depth.cc | 95 type::SpaceToDepth(op_params, GetTensorShape(input), \ in Eval()
|
| /external/tensorflow/tensorflow/lite/delegates/hexagon/ |
| D | README.md | 101 * SpaceToDepth
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/ |
| D | tpu_space_to_depth_pass.mlir | 18 …// CHECK-DAG: %[[SPACETODEPTH0:.*]] = "tf.SpaceToDepth"([[INPUT:.*]]) {block_size = 2 : i64, data_…
93 …// CHECK-DAG: %[[SPACETODEPTH00:.*]] = "tf.SpaceToDepth"([[INPUT00:.*]]#0) {block_size = 2 : i64, …
96 …// CHECK-DAG: %[[SPACETODEPTH01:.*]] = "tf.SpaceToDepth"([[INPUT01:.*]]#0) {block_size = 2 : i64, …
170 …// CHECK-DAG: %[[SPACETODEPTH00:.*]] = "tf.SpaceToDepth"([[INPUT00:.*]]#0) {block_size = 2 : i64, …
|
| /external/tensorflow/tensorflow/cc/gradients/ |
| D | array_grad_test.cc | 317 auto y = SpaceToDepth(scope_, x, /* block_size */ 2); in TEST_F()
|
| D | array_grad.cc | 386 grad_outputs->push_back(SpaceToDepth(scope, grad_inputs[0], block_size)); in DepthToSpaceGrad()
|
| /external/tensorflow/tensorflow/lite/g3doc/guide/ |
| D | op_select_allowlist.md | 494 * `raw_ops.SpaceToDepth`
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes_test.cc | 6735 TestConvertDepthSpaceShuffle<ops::SpaceToDepth, dtype, CType>(test, params); in TEST_P()
6743 GetDepthSpaceShuffleNodeDef<ops::SpaceToDepth>(DT_FLOAT, 2, "NCHW"); in TEST_P()
6753 GetDepthSpaceShuffleNodeDef<ops::SpaceToDepth>(DT_FLOAT, 2, "NCHW"); in TEST_P()
6763 GetDepthSpaceShuffleNodeDef<ops::SpaceToDepth>(DT_FLOAT, 3, "NCHW"); in TEST_P()
6773 GetDepthSpaceShuffleNodeDef<ops::SpaceToDepth>(DT_FLOAT, 3, "NCHW"); in TEST_P()
6782 NodeDef node_def = GetDepthSpaceShuffleNodeDef<ops::SpaceToDepth>( in TEST_P()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/g3doc/ |
| D | gpu_supported_ops.md | 220 `SpaceToDepth` | `T={bool,complex64,double,float,int32,int64,uint32,uint64}`
|
| D | cpu_supported_ops.md | 223 `SpaceToDepth` | `T={bool,complex64,double,float,int32,int64,uint32,uint64}`
|
| /external/tensorflow/tensorflow/lite/toco/tflite/ |
| D | operator.cc | 772 class SpaceToDepth class
1890 ops.push_back(MakeUnique<SpaceToDepth>( in BuildOperatorList()
|
| /external/tensorflow/tensorflow/compiler/mlir/tosa/tests/ |
| D | tf-to-tosa-pipeline.mlir | 761 …%2 = "tf.SpaceToDepth"(%arg0) {block_size = 2 : i64, data_format = "NHWC"} : (tensor<1x32x32x8xf…
|
| /external/tensorflow/tensorflow/compiler/mlir/tosa/g3doc/ |
| D | legalization.md | 2763 ### tf.SpaceToDepth
2765 SpaceToDepth for tensors of type T.
2770 %output = tf.SpaceToDepth(%input) {block_size, data_format}
|
| /external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/ |
| D | legalize_tf.cc | 122 DECL_CONVERT_OP(SpaceToDepth);
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | tooling_util.cc | 339 HANDLE_OPERATORTYPENAME_CASE(SpaceToDepth) in OperatorTypeName()
|