HomeSort by relevance Sort by last modified time
    Searched defs:input_strides (Results 1 - 3 of 3) sorted by null

  /external/tensorflow/tensorflow/contrib/lite/toco/graph_transformations/
resolve_constant_transpose.cc 59 int input_strides[4]; local
60 input_strides[3] = 1;
61 input_strides[2] = extended_input_dims[3];
62 input_strides[1] = input_strides[2] * extended_input_dims[2];
63 input_strides[0] = input_strides[1] * extended_input_dims[1];
64 const int input_stride_0 = input_strides[extended_perm[3]];
65 const int input_stride_1 = input_strides[extended_perm[2]];
66 const int input_stride_2 = input_strides[extended_perm[1]]
    [all...]
  /external/eigen/unsupported/Eigen/CXX11/src/Tensor/
TensorReduction.h 448 array<Index, NumInputDims> input_strides; local
450 input_strides[0] = 1;
452 input_strides[i] = input_strides[i-1] * input_dims[i-1];
455 input_strides.back() = 1;
457 input_strides[i] = input_strides[i + 1] * input_dims[i + 1];
465 m_reducedStrides[reduceIndex] = input_strides[i];
468 m_preservedStrides[outputIndex] = input_strides[i];
    [all...]
  /external/tensorflow/tensorflow/contrib/lite/toco/
tooling_util.cc 1694 int input_strides[4]; local
    [all...]

Completed in 144 milliseconds