HomeSort by relevance Sort by last modified time
    Searched refs:DepthToSpace (Results 1 - 8 of 8) sorted by null

  /frameworks/ml/nn/common/operations/
Reshape.cpp 58 tflite::optimized_ops::DepthToSpace(
65 tflite::optimized_ops::DepthToSpace(
  /external/tensorflow/tensorflow/contrib/lite/toco/tflite/
operator.cc 239 class DepthToSpace : public CustomOperator<DepthToSpaceOperator> {
834 new DepthToSpace("DEPTH_TO_SPACE", OperatorType::kDepthToSpace));
    [all...]
  /external/tensorflow/tensorflow/cc/gradients/
array_grad.cc 348 grad_outputs->push_back(DepthToSpace(scope, grad_inputs[0], block_size));
362 REGISTER_GRADIENT_OP("DepthToSpace", DepthToSpaceGrad);
array_grad_test.cc 317 auto y = DepthToSpace(scope_, x, /* block_size */ 2);
  /external/tensorflow/tensorflow/contrib/lite/toco/
tooling_util.cc 248 HANDLE_OPERATORTYPENAME_CASE(DepthToSpace)
    [all...]
  /external/tensorflow/tensorflow/compiler/tests/
randomized_tests.cc     [all...]
  /external/tensorflow/tensorflow/contrib/lite/kernels/internal/reference/
reference_ops.h 364 inline void DepthToSpace(const T* input_data, const Dims<4>& input_dims,
    [all...]
  /external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/
optimized_ops.h     [all...]

Completed in 1140 milliseconds