Searched defs:DepthToSpace (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/lite/toco/tflite/ |
| D | operator.cc | 391 class DepthToSpace : public CustomOperator<DepthToSpaceOperator> { class
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | legacy_optimized_ops.h | 1589 inline void DepthToSpace(const T* input_data, const Dims<4>& input_dims, in DepthToSpace() function
|
| D | optimized_ops.h | 2470 inline void DepthToSpace(const tflite::DepthToSpaceParams& op_params, in DepthToSpace() function
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | legacy_reference_ops.h | 1754 inline void DepthToSpace(const T* input_data, const Dims<4>& input_dims, in DepthToSpace() function
|
| D | reference_ops.h | 290 inline void DepthToSpace(const tflite::DepthToSpaceParams& op_params, in DepthToSpace() function
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | randomized_tests.cc | 1761 TEST_F(OpTest, DepthToSpace) { in TEST_F() argument
|
| /external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 2638 inline void DepthToSpace(const tflite::DepthToSpaceParams& op_params, in DepthToSpace() function
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 8976 func DepthToSpace(scope *Scope, input tf.Output, block_size int64, optional ...DepthToSpaceAttr) (o… func
|