Searched defs:tensor_types (Results 1 – 3 of 3) sorted by relevance
302 static std::vector<PyTensorType*> tensor_types; variable346 static void initialize_aten_types(std::vector<PyTensorType*>& tensor_types) { in initialize_aten_types()392 const std::vector<PyTensorType*>& tensor_types) { in py_bind_tensor_types()
159 std::vector<TensorTypePtr> tensor_types; in gatherTensorTypes() local771 std::optional<at::ScalarType> t) -> TensorTypePtr { in PropagateTensorShapeOnNode()1710 std::vector<TensorTypePtr> tensor_types; in PropagateTensorShapeOnNode() local1714 const std::vector<TensorTypePtr>& tensor_types) -> TensorTypePtr { in PropagateTensorShapeOnNode()1875 std::vector<TensorTypePtr> tensor_types) { in PropagateCompleteShapeOnNode()
217 llvm::SmallVector<mlir::TensorType> tensor_types; in GetTensorType() local