Searched defs:DepthToSpace (Results 1 – 8 of 8) sorted by relevance
| /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/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
|
| /external/tensorflow/tensorflow/lite/toco/tflite/ |
| D | operator.cc | 392 class DepthToSpace class
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | randomized_tests.cc | 1762 TEST_F(OpTest, DepthToSpace) { in TEST_F() argument
|
| /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
|
| D | optimized_ops.h | 1641 inline void DepthToSpace(const tflite::DepthToSpaceParams& op_params, in DepthToSpace() function
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 1110 func DepthToSpace(scope *Scope, input tf.Output, block_size int64, optional ...DepthToSpaceAttr) (o… func
|