Searched defs:GetDeviceOrdinal (Results 1 – 5 of 5) sorted by relevance
145 StatusOr<int> StreamExecutorTransferOpImpl::GetDeviceOrdinal( in GetDeviceOrdinal() function in tensorflow::StreamExecutorTransferOpImpl
58 StatusOr<mlir::Value> GetDeviceOrdinal(const Mesh& mesh, in GetDeviceOrdinal() function
69 llvm::Optional<int64_t> GetDeviceOrdinal( in GetDeviceOrdinal() function
261 int GetDeviceOrdinal(const xla::ExecutableRunOptions* run_options) { in GetDeviceOrdinal() function
961 absl::optional<uint32> GetDeviceOrdinal() { return device_ordinal_; } in GetDeviceOrdinal() function in tensorflow::profiler::__anon2fb572530111::ScopedCudaContext