Searched refs:tensorId (Results 1 – 2 of 2) sorted by relevance
| /external/llvm-project/mlir/tools/mlir-linalg-ods-gen/ |
| D | mlir-linalg-ods-gen.cpp | 866 : Expression(Kind::TensorUse), tensorId(name), indexingMap(map) {} in TensorUse()
874 return tensorId == other.tensorId && indexingMap == other.indexingMap; in operator ==()
882 StringRef tensorId; member
1036 return ::llvm::hash_value(val.tensorId); // don't care about collisions. in getHashValue()
1125 StringRef tensorId = parser.curToken.getSpelling(); in parseTensorDef() local
1141 tensorId, RegisteredTensor{tensorType, map, isOutput, AffineMap(), in parseTensorDef()
1145 LLVM_DEBUG(llvm::dbgs() << "Recorded: " << tensorId << " " in parseTensorDef()
1159 StringRef tensorId = parser.curToken.getSpelling(); in parseTensorUse() local
1166 LLVM_DEBUG(llvm::dbgs() << "Use of tensor: " << tensorId << " map: " << map in parseTensorUse()
1169 result = TensorUse(tensorId, map); in parseTensorUse()
[all …]
|
| /external/tensorflow/tensorflow/lite/kernels/hashtable/ |
| D | hashtable_ops_test.cc | 69 void SetTensorData(Interpreter* interpreter, int tensorId, in SetTensorData() argument
71 auto* tensor = interpreter->tensor(tensorId); in SetTensorData()
80 void SetTensorData(Interpreter* interpreter, int tensorId, in SetTensorData() argument
82 auto* tensor = interpreter->tensor(tensorId); in SetTensorData()
|