| /external/eigen/unsupported/Eigen/CXX11/src/Tensor/ |
| D | TensorChipping.h | 24 template<DenseIndex DimId, typename XprType>
25 struct traits<TensorChippingOp<DimId, XprType> > : public traits<XprType>
37 template<DenseIndex DimId, typename XprType>
38 struct eval<TensorChippingOp<DimId, XprType>, Eigen::Dense>
40 typedef const TensorChippingOp<DimId, XprType>& type;
43 template<DenseIndex DimId, typename XprType>
44 struct nested<TensorChippingOp<DimId, XprType>, 1, typename eval<TensorChippingOp<DimId, XprType> >…
46 typedef TensorChippingOp<DimId, XprType> type;
49 template <DenseIndex DimId>
53 eigen_assert(dim == DimId);
[all …]
|
| D | TensorBase.h | 752 template <Index DimId> EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE
753 const TensorChippingOp<DimId, const Derived>
755 return TensorChippingOp<DimId, const Derived>(derived(), offset, DimId); in chip()
940 template <DenseIndex DimId> EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE
941 const TensorChippingOp<DimId, const Derived>
943 return TensorChippingOp<DimId, const Derived>(derived(), offset, DimId); in chip()
945 template <Index DimId> EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE
946 TensorChippingOp<DimId, Derived>
948 return TensorChippingOp<DimId, Derived>(derived(), offset, DimId); in chip()
|
| D | TensorForwardDeclarations.h | 48 template<DenseIndex DimId, typename XprType> class TensorChippingOp;
|
| /external/llvm-project/mlir/include/mlir/Dialect/SDBM/ |
| D | SDBMExpr.h | 26 enum class SDBMExprKind { Add, Stripe, Diff, Constant, DimId, SymbolId, Neg }; enumerator
184 return expr.getKind() == SDBMExprKind::DimId || in isClassFor()
209 return expr.getKind() == SDBMExprKind::DimId || in isClassFor()
226 return expr.getKind() == SDBMExprKind::DimId || in isClassFor()
299 return expr.getKind() == SDBMExprKind::DimId || in isClassFor()
318 return expr.getKind() == SDBMExprKind::DimId; in isClassFor()
365 case SDBMExprKind::DimId: in visit()
421 if (expr.getKind() == SDBMExprKind::DimId) in visitInput()
|
| /external/llvm-project/mlir/lib/IR/ |
| D | AffineExpr.cpp | 67 case AffineExprKind::DimId: { in replaceDimsAndSymbols()
140 case AffineExprKind::DimId: in isSymbolicOrConstant()
163 case AffineExprKind::DimId: in isPureAffine()
196 case AffineExprKind::DimId: in getLargestKnownDivisor()
225 case AffineExprKind::DimId: in isMultipleOf()
254 if (getKind() == AffineExprKind::DimId) { in isFunctionOfDim()
296 case AffineExprKind::DimId: in isDivisibleBySymbol()
357 case AffineExprKind::DimId: in symbolicDivide()
406 case AffineExprKind::DimId: in simplifySemiAffine()
454 return getAffineDimOrSymbol(AffineExprKind::DimId, position, context); in getAffineDimExpr()
|
| D | AffineMap.cpp | 60 case AffineExprKind::DimId: in constantFoldImpl()
|
| D | AsmPrinter.cpp | 1996 case AffineExprKind::DimId: { in printAffineExprInternal()
|
| /external/llvm-project/mlir/include/mlir/IR/ |
| D | AffineExprVisitor.h | 105 case AffineExprKind::DimId: in walkPostOrder()
142 case AffineExprKind::DimId: in visit()
|
| D | AffineExpr.h | 57 DimId, enumerator
252 return getKind() == AffineExprKind::DimId; in isa()
|
| /external/llvm-project/polly/lib/Transform/ |
| D | ScheduleOptimizer.cpp | 563 isl::id DimId; in permuteDimensions() local
565 DimId = Map.get_tuple_id(DimType); in permuteDimensions()
576 if (DimId) in permuteDimensions()
577 Map = Map.set_tuple_id(DimType, DimId); in permuteDimensions()
|
| /external/llvm-project/mlir/lib/Dialect/Linalg/Transforms/ |
| D | Bufferize.cpp | 92 case AffineExprKind::DimId: { in allocateBuffersForResults()
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | graph_properties.cc | 1124 struct DimId { struct in tensorflow::grappler::SymbolicShapeRefiner
1128 bool operator==(const DimId& other) const { in operator ==() argument
1135 std::size_t operator()(const DimId& dim) const { in operator ()()
1368 DimId id{node, index, dim_id}; in GetUnknownOutputDim()
2003 absl::flat_hash_map<DimId, DimensionHandle, HashDimId> unknown_dims_;
|
| /external/llvm-project/mlir/lib/Dialect/SDBM/ |
| D | SDBMExpr.cpp | 610 assignDialect, static_cast<unsigned>(SDBMExprKind::DimId), position); in get()
|
| /external/swiftshader/third_party/llvm-10.0/llvm/lib/Target/AMDGPU/AsmParser/ |
| D | AMDGPUAsmParser.cpp | 6511 StringRef DimId = Token; in parseDim() local
6512 if (DimId.startswith("SQ_RSRC_IMG_")) in parseDim()
6513 DimId = DimId.substr(12); in parseDim()
6515 const AMDGPU::MIMGDimInfo *DimInfo = AMDGPU::getMIMGDimInfoByAsmSuffix(DimId); in parseDim()
|
| /external/llvm-project/llvm/lib/Target/AMDGPU/AsmParser/ |
| D | AMDGPUAsmParser.cpp | 7326 StringRef DimId = Token; in parseDim() local
7327 if (DimId.startswith("SQ_RSRC_IMG_")) in parseDim()
7328 DimId = DimId.substr(12); in parseDim()
7330 const AMDGPU::MIMGDimInfo *DimInfo = AMDGPU::getMIMGDimInfoByAsmSuffix(DimId); in parseDim()
|