Searched defs:DepthToSpace (Results 1 - 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/contrib/lite/toco/tflite/
H A Doperator.cc239 class DepthToSpace : public CustomOperator<DepthToSpaceOperator> { class in namespace:toco::tflite
834 new DepthToSpace("DEPTH_TO_SPACE", OperatorType::kDepthToSpace));
/external/tensorflow/tensorflow/contrib/lite/kernels/internal/reference/
H A Dreference_ops.h364 inline void DepthToSpace(const T* input_data, const Dims<4>& input_dims, function in namespace:tflite::reference_ops
/external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/
H A Doptimized_ops.h961 inline void DepthToSpace(const T* input_data, const Dims<4>& input_dims, function in namespace:tflite::optimized_ops
964 gemmlowp::ScopedProfilingLabel label("DepthToSpace");

Completed in 229 milliseconds