Home
last modified time | relevance | path

Searched defs:GetDeviceOrdinal (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/tpu/kernels/
H A Dtransfer_ops.cc145 StatusOr<int> StreamExecutorTransferOpImpl::GetDeviceOrdinal( in GetDeviceOrdinal() function in tensorflow::StreamExecutorTransferOpImpl
/aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/
H A Ddtensor_send_recv.cc58 StatusOr<mlir::Value> GetDeviceOrdinal(const Mesh& mesh, in GetDeviceOrdinal() function
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
H A Dreplicate_to_island.cc69 llvm::Optional<int64_t> GetDeviceOrdinal( in GetDeviceOrdinal() function
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/
H A Dcpu_runtime.cc261 int GetDeviceOrdinal(const xla::ExecutableRunOptions* run_options) { in GetDeviceOrdinal() function
/aosp_15_r20/external/tensorflow/tensorflow/core/profiler/backends/gpu/
H A Dcupti_tracer.cc961 absl::optional<uint32> GetDeviceOrdinal() { return device_ordinal_; } in GetDeviceOrdinal() function in tensorflow::profiler::__anon2fb572530111::ScopedCudaContext