Searched refs:MPIType (Results 1 – 5 of 5) sorted by relevance
| /external/clang/lib/StaticAnalyzer/Checkers/MPI-Checker/ |
| D | MPIFunctionClassifier.cpp | 33 MPIType.push_back(IdentInfo_MPI_Send); in initPointToPointIdentifiers()
39 MPIType.push_back(IdentInfo_MPI_Isend); in initPointToPointIdentifiers()
44 MPIType.push_back(IdentInfo_MPI_Ssend); in initPointToPointIdentifiers()
50 MPIType.push_back(IdentInfo_MPI_Issend); in initPointToPointIdentifiers()
55 MPIType.push_back(IdentInfo_MPI_Bsend); in initPointToPointIdentifiers()
61 MPIType.push_back(IdentInfo_MPI_Ibsend); in initPointToPointIdentifiers()
66 MPIType.push_back(IdentInfo_MPI_Rsend); in initPointToPointIdentifiers()
71 MPIType.push_back(IdentInfo_MPI_Irsend); in initPointToPointIdentifiers()
76 MPIType.push_back(IdentInfo_MPI_Recv); in initPointToPointIdentifiers()
82 MPIType.push_back(IdentInfo_MPI_Irecv); in initPointToPointIdentifiers()
[all …]
|
| D | MPIFunctionClassifier.h | 68 llvm::SmallVector<IdentifierInfo *, 32> MPIType; variable
|
| /external/tensorflow/tensorflow/contrib/mpi_collectives/kernels/ |
| D | ring.h | 47 MPI_Datatype MPIType();
208 MPIType<T>(), recv_from, TAG_TENSOR, in RingAllreduce()
212 MPIType<T>(), send_to, TAG_TENSOR, in RingAllreduce()
240 segment_send, segment_sizes[send_seg_id], MPIType<T>(), send_to, in RingAllreduce()
241 TAG_TENSOR, segment_recv, segment_sizes[recv_seg_id], MPIType<T>(), in RingAllreduce()
312 segment_send, elements_per_row * rows_send, MPIType<T>(), send_to, in RingAllgather()
313 TAG_TENSOR, segment_recv, elements_per_row * rows_recv, MPIType<T>(), in RingAllgather()
|
| D | ring.cc | 28 extern template MPI_Datatype MPIType<float>();
29 extern template MPI_Datatype MPIType<int>();
30 extern template MPI_Datatype MPIType<long long>();
|
| D | ring.cu.cc | 31 MPI_Datatype MPIType<float>() { in MPIType() function
35 MPI_Datatype MPIType<int>() { in MPIType() function
39 MPI_Datatype MPIType<long long>() { in MPIType() function
|