| /external/tensorflow/tensorflow/lite/delegates/xnnpack/ |
| D | depth_to_space_test.cc | 29 TEST(DepthToSpace, SinglePixel) { in TEST() argument
52 TEST(DepthToSpace, SingleRow) { in TEST() argument
77 TEST(DepthToSpace, SingleColumn) { in TEST() argument
102 TEST(DepthToSpace, FullImage) { in TEST() argument
127 TEST(DepthToSpace, MultiThreading) { in TEST() argument
|
| /external/tensorflow/tensorflow/core/api_def/java_api/ |
| D | api_def_DepthToSpace.pbtxt | 2 graph_op_name: "DepthToSpace"
4 name: "nn.DepthToSpace"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | DepthToSpace.pbtxt | 2 name: "DepthToSpace"
23 name: "DepthToSpace"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | DepthToSpace.pbtxt | 2 name: "DepthToSpace"
23 name: "DepthToSpace"
|
| /external/tensorflow/tensorflow/core/api_def/python_api/ |
| D | api_def_DepthToSpace.pbtxt | 2 graph_op_name: "DepthToSpace"
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_DepthToSpace.pbtxt | 2 graph_op_name: "DepthToSpace"
9 summary: "DepthToSpace for tensors of type T."
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | depth_to_space.h | 24 inline void DepthToSpace(const tflite::DepthToSpaceParams& op_params, in DepthToSpace() function
|
| D | legacy_reference_ops.h | 1845 inline void DepthToSpace(const T* input_data, const Dims<4>& input_dims, in DepthToSpace() function
1851 DepthToSpace(op_params, DimsToShape(input_dims), input_data, in DepthToSpace()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | depth_to_space.cc | 99 type::DepthToSpace(op_params, GetTensorShape(input), \ in Eval()
|
| /external/tensorflow/tensorflow/lite/micro/kernels/ |
| D | depth_to_space.cc | 99 type::DepthToSpace(op_params, GetTensorShape(input), \ in Eval()
|
| /external/tensorflow/tensorflow/lite/delegates/hexagon/ |
| D | README.md | 70 * DepthToSpace
|
| /external/tensorflow/tensorflow/cc/gradients/ |
| D | array_grad_test.cc | 325 auto y = DepthToSpace(scope_, x, /* block_size */ 2); in TEST_F()
|
| D | array_grad.cc | 375 grad_outputs->push_back(DepthToSpace(scope, grad_inputs[0], block_size)); in SpaceToDepthGrad()
|
| /external/tensorflow/tensorflow/lite/g3doc/guide/ |
| D | op_select_allowlist.md | 139 * `raw_ops.DepthToSpace`
|
| /external/tensorflow/tensorflow/compiler/tf2xla/g3doc/ |
| D | gpu_supported_ops.md | 63 `DepthToSpace` | `T={bool,complex64,double,float,int32,int64,uint32,uint64}`
|
| D | cpu_supported_ops.md | 63 `DepthToSpace` | `T={bool,complex64,double,float,int32,int64,uint32,uint64}`
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes_test.cc | 6636 TestConvertDepthSpaceShuffle<ops::DepthToSpace, dtype, CType>(test, params); in TEST_P()
6644 GetDepthSpaceShuffleNodeDef<ops::DepthToSpace>(DT_FLOAT, 2, "NCHW"); in TEST_P()
6654 GetDepthSpaceShuffleNodeDef<ops::DepthToSpace>(DT_FLOAT, 2, "NCHW"); in TEST_P()
6664 GetDepthSpaceShuffleNodeDef<ops::DepthToSpace>(DT_FLOAT, 3, "NCHW"); in TEST_P()
6673 NodeDef node_def = GetDepthSpaceShuffleNodeDef<ops::DepthToSpace>( in TEST_P()
|
| /external/tensorflow/tensorflow/lite/toco/tflite/ |
| D | operator.cc | 392 class DepthToSpace class
1892 ops.push_back(MakeUnique<DepthToSpace>( in BuildOperatorList()
|
| /external/tensorflow/tensorflow/compiler/mlir/tosa/tests/ |
| D | tf-to-tosa-pipeline.mlir | 773 …%2 = "tf.DepthToSpace"(%arg0) {block_size = 2 : i64, data_format = "NHWC"} : (tensor<1x32x32x8xf…
|
| /external/tensorflow/tensorflow/compiler/mlir/tosa/g3doc/ |
| D | legalization.md | 1563 ### tf.DepthToSpace
1570 %output = tf.DepthToSpace(%input) {block_size, data_format}
|
| /external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/ |
| D | legalize_tf.cc | 123 DECL_CONVERT_OP(DepthToSpace);
|
| D | legalize_tfl.cc | 129 DECL_CONVERT_OP(DepthToSpace);
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | tooling_util.cc | 338 HANDLE_OPERATORTYPENAME_CASE(DepthToSpace) in OperatorTypeName()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | legacy_optimized_ops.h | 4645 inline void DepthToSpace(const T* input_data, const Dims<4>& input_dims, in DepthToSpace() function
4651 DepthToSpace(op_params, DimsToShape(input_dims), input_data, in DepthToSpace()
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | randomized_tests.cc | 1762 TEST_F(OpTest, DepthToSpace) { in TEST_F() argument
|