Home / Class/ make_ideep_tensor Class — pytorch Architecture

make_ideep_tensor Class — pytorch Architecture

Architecture documentation for the make_ideep_tensor class in Matmul.cpp from the pytorch codebase.

Entity Profile

Source Code

aten/src/ATen/native/mkldnn/Matmul.cpp lines 131–144

template <typename scalar_t>
inline ideep::tensor make_ideep_tensor(
    std::vector<int64_t> dims,
    ideep::tensor::data_type idtype,
    ideep::tensor::dims& strides,
    scalar_t *data){
    ideep::tensor res({
      dims,
      idtype,
      strides
      },
    data);
    return res;
  }

Analyze Your Own Codebase

Get architecture documentation, dependency graphs, and domain analysis for your codebase in minutes.

Try Supermodel Free