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

  /external/tensorflow/tensorflow/core/kernels/
depthtospace_op.cc 76 constexpr int kDims = is_int8x4 ? 5 : 4;
77 OP_REQUIRES(context, kDims == dims,
78 errors::InvalidArgument("Input rank should be: ", kDims,
112 auto Tinput = input.tensor<T, kDims>();
113 auto Toutput = outputs_tensor->tensor<T, kDims>();
spacetodepth_op.cc 76 constexpr int kDims = is_int8x4 ? 5 : 4;
77 OP_REQUIRES(context, kDims == dims,
78 errors::InvalidArgument("Input rank should be: ", kDims,
114 auto Tinput = input.tensor<T, kDims>();
115 auto Toutput = outputs_tensor->tensor<T, kDims>();
  /external/tensorflow/tensorflow/contrib/lite/toco/
export_tensorflow.cc 96 const int kDims = input_shape.dimensions_count();
98 kDims > 1 || (kDims == 1 && input_shape.dims(0) > 1)) {
99 for (int i = 0; i < kDims; ++i) {
    [all...]

Completed in 874 milliseconds