Searched defs:SpaceToDepth (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/lite/delegates/gpu/gl/kernels/ |
| D | space_to_depth.cc | 32 class SpaceToDepth : public NodeShader { class
|
| /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
|
| /external/tensorflow/tensorflow/lite/toco/tflite/ |
| D | operator.cc | 772 class SpaceToDepth class
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | randomized_tests.cc | 3039 TEST_F(OpTest, SpaceToDepth) { in TEST_F() argument
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | legacy_optimized_ops.h | 4656 inline void SpaceToDepth(const T* input_data, const Dims<4>& input_dims, in SpaceToDepth() function
|
| D | optimized_ops.h | 1682 inline void SpaceToDepth(const tflite::SpaceToDepthParams& op_params, in SpaceToDepth() function
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 1229 func SpaceToDepth(scope *Scope, input tf.Output, block_size int64, optional ...SpaceToDepthAttr) (o… func
|