Home
last modified time | relevance | path

Searched defs:dim_type (Results 1 – 3 of 3) sorted by relevance

/external/pytorch/aten/src/ATen/native/cuda/
DSpectralOps.cu21 using dim_type = std::remove_cv_t<decltype(MAX_DIMS)>; typedef
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ops/
Deinsum.cc80 EinsumDimensionType dim_type, in OrderMatches()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dlegalize_hlo.cc1889 auto dim_type = RankedTensorType::get( in matchAndRewrite() local
2073 auto dim_type = RankedTensorType::get({1}, rewriter.getI64Type()); in matchAndRewrite() local