| /external/tensorflow/tensorflow/stream_executor/tpu/ |
| D | tpu_topology.cc | 25 tpu::ExecutorApiFn()->TpuCoreLocation_ChipCoordinatesFn(core_location_, &x, in chip_coordinates()
32 tpu::ExecutorApiFn()->TpuCoreLocation_HostCoordinatesFn(core_location_, &x, in host_coordinates()
38 return tpu::ExecutorApiFn()->TpuCoreLocation_IndexFn(core_location_); in index()
42 return tpu::ExecutorApiFn()->TpuCoreLocation_IdFn(core_location_); in Id()
46 return tpu::ExecutorApiFn()->TpuHostLocation_IdFn(host_location_); in Id()
51 int num_cores = tpu::ExecutorApiFn()->TpuHostLocation_NumCoresFn( in Cores()
54 tpu::ExecutorApiFn()->TpuHostLocation_CoresFn(host_location_, core_type, in Cores()
66 return tpu::ExecutorApiFn()->TpuTopology_LogicalDevicesPerHostFn(topology_, in LogicalDevicesPerHost()
72 return tpu::ExecutorApiFn()->TpuTopology_LogicalDevicesPerChipFn(topology_, in LogicalDevicesPerChip()
77 return tpu::ExecutorApiFn()->TpuTopology_HostCountFn(topology_); in HostCount()
[all …]
|
| D | tpu_executor.cc | 36 tpu::ExecutorApiFn()->TpuExecutor_FreeFn(executor_); in ~TpuExecutor()
43 tpu::ExecutorApiFn()->TpuExecutor_NewDeviceOptionsFn( in Init()
45 tpu::ExecutorApiFn()->TpuExecutor_InitFn(executor_, device_ordinal, options, in Init()
47 tpu::ExecutorApiFn()->TpuExecutor_FreeDeviceOptionsFn(options); in Init()
52 return tpu::ExecutorApiFn()->TpuExecutor_PlatformDeviceCountFn(executor_); in PlatformDeviceCount()
56 tpu::ExecutorApiFn()->TpuExecutor_SyncAndForgetFailedStreamsFn(executor_); in SyncAndForgetFailedStreams()
60 return tpu::ExecutorApiFn()->TpuExecutor_SynchronizeAllActivityFn(executor_); in SynchronizeAllActivity()
65 tpu::ExecutorApiFn()->TpuExecutor_BlockHostUntilDoneFn( in BlockHostUntilDone()
72 tpu::ExecutorApiFn()->TpuExecutor_BlockUntilDoneOrFailedFn(executor_, in BlockUntilDoneOrFailed()
79 tpu::ExecutorApiFn()->TpuExecutor_GetStatusFn( in GetStatus()
[all …]
|
| D | tpu_platform.cc | 36 platform_ = tpu::ExecutorApiFn()->TpuPlatform_NewFn(); in TpuPlatform()
60 tpu::ExecutorApiFn()->TpuPlatform_InitializeFn( in Initialize()
70 return tpu::ExecutorApiFn()->TpuPlatform_InitializedFn(platform_); in Initialized()
74 tpu::ExecutorApiFn()->TpuPlatform_FreeFn(platform_); in ~TpuPlatform()
78 return tpu::ExecutorApiFn()->TpuPlatform_VisibleDeviceCountFn(platform_); in VisibleDeviceCount()
91 tpu::ExecutorApiFn()->TpuStreamExecutorConfig_DefaultFn(); in GetUncachedExecutor()
93 tpu::ExecutorApiFn()->TpuStreamExecutorConfig_SetOrdinalFn(c_config, in GetUncachedExecutor()
97 SE_StreamExecutor* executor = tpu::ExecutorApiFn()->TpuPlatform_GetExecutorFn( in GetUncachedExecutor()
99 tpu::ExecutorApiFn()->TpuStreamExecutorConfig_FreeFn(c_config); in GetUncachedExecutor()
114 return tpu::ExecutorApiFn()->TpuPlatform_TpuMemoryLimitFn(platform_); in TpuMemoryLimit()
[all …]
|
| D | tpu_transfer_manager.cc | 41 manager_ = tpu::ExecutorApiFn()->TpuTransferManager_NewFn(); in TpuTransferManager()
45 tpu::ExecutorApiFn()->TpuTransferManager_FreeFn(manager_); in ~TpuTransferManager()
59 tpu::ExecutorApiFn()->TpuTransferManager_HostShapeToDeviceShapeFn( in HostShapeToDeviceShape()
79 tpu::ExecutorApiFn()->TpuTransferManager_TransferLiteralToDeviceAsyncFn( in TransferLiteralToDeviceAsync()
97 tpu::ExecutorApiFn()->TpuTransferManager_TransferLiteralToInfeedFn( in TransferLiteralToInfeed()
123 tpu::ExecutorApiFn()->TpuTransferManager_TransferBuffersToInfeedFn( in TransferBuffersToInfeed()
140 tpu::ExecutorApiFn()->TpuTransferManager_TransferLiteralFromOutfeedFn( in TransferLiteralFromOutfeed()
160 tpu::ExecutorApiFn()->TpuTransferManager_ResetDevicesFn( in ResetDevices()
168 tpu::ExecutorApiFn()->TpuStatus_NewFn(); // OK or the first error
172 if (!tpu::ExecutorApiFn()->TpuStatus_OkFn(status) && in TransferFinished()
[all …]
|
| D | status_helper.h | 26 : c_status(tensorflow::tpu::ExecutorApiFn()->TpuStatus_NewFn()) {} in StatusHelper()
29 tensorflow::tpu::ExecutorApiFn()->TpuStatus_FreeFn(c_status); in ~StatusHelper()
33 if (tensorflow::tpu::ExecutorApiFn()->TpuStatus_OkFn(c_status)) { in FromC()
38 tensorflow::tpu::ExecutorApiFn()->TpuStatus_CodeFn(c_status)), in FromC()
39 tensorflow::tpu::ExecutorApiFn()->TpuStatus_MessageFn(c_status)); in FromC()
44 return tensorflow::tpu::ExecutorApiFn()->TpuStatus_OkFn(c_status); in ok()
|
| D | tpu_stream.h | 35 tensorflow::tpu::ExecutorApiFn()->TpuStream_FreeFn(stream_); in ~TpuStream()
40 return tensorflow::tpu::ExecutorApiFn() in IsSameSharedMemoryLocation()
49 tensorflow::tpu::ExecutorApiFn()->TpuStream_EnqueueTransferHostToDeviceFn( in EnqueueTransferHostToDevice()
59 tensorflow::tpu::ExecutorApiFn()->TpuStream_EnqueueTransferDeviceToHostFn( in EnqueueTransferDeviceToHost()
69 tensorflow::tpu::ExecutorApiFn() in EnqueueOnTpuDeviceSendRecvLocal()
|
| D | tpu_timer.h | 30 tensorflow::tpu::ExecutorApiFn()->TpuTimer_FreeFn(timer_); in ~TpuTimer()
33 return tensorflow::tpu::ExecutorApiFn()->TpuTimer_MicrosecondsFn(timer_); in Microseconds()
36 return tensorflow::tpu::ExecutorApiFn()->TpuTimer_NanosecondsFn(timer_); in Nanoseconds()
|
| D | tpu_event.h | 29 tensorflow::tpu::ExecutorApiFn()->TpuEvent_FreeFn(event_); in ~TpuEvent()
|
| D | c_api_conversions.cc | 99 tensorflow::tpu::ExecutorApiFn()->TpuStatus_SetFn( in ToC()
114 tensorflow::tpu::ExecutorApiFn()->TpuStatus_SetFn( in ToC()
|
| /external/tensorflow/tensorflow/core/tpu/ |
| D | tpu_on_demand_compiler.cc | 86 using ::tensorflow::tpu::ExecutorApiFn;
96 ExecutorApiFn()->TpuExecutable_FreeFn(se_executable_); in ~TpuExecutable()
134 ExecutorApiFn()->TpuExecutable_ExecuteAsyncOnStreamFn( in ExecuteAsyncOnStream()
165 ExecutorApiFn()->TpuExecutable_FreeXlaShapeIndexArrayFn( in ExecuteAsyncOnStream()
175 ExecutorApiFn()->TpuExecutable_FreeMaybeOwningDeviceMemoryArrayFn( in ExecuteAsyncOnStream()
184 ExecutorApiFn()->TpuExecutable_FingerprintFn(se_executable_, &data, &size); in fingerprint()
211 TpuCompiler() { compiler_ = ExecutorApiFn()->TpuCompiler_NewFn(); } in TpuCompiler()
212 ~TpuCompiler() override { ExecutorApiFn()->TpuCompiler_FreeFn(compiler_); } in ~TpuCompiler()
232 ExecutorApiFn()->TpuCompiler_RunHloPassesFn( in RunHloPasses()
261 ExecutorApiFn()->TpuCompiler_RunBackendFn( in RunBackend()
[all …]
|
| D | tpu_executor_api.h | 25 TfTpu_ExecutorApiFn* ExecutorApiFn();
|
| D | tpu_executor_api.cc | 21 TfTpu_ExecutorApiFn* ExecutorApiFn() { in ExecutorApiFn() function
|
| D | tpu_executor_init_fns.inc | 4 auto* executor_fn = tensorflow::tpu::ExecutorApiFn();
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | tpu_computation_placer.cc | 30 placer_ = tensorflow::tpu::ExecutorApiFn()->TpuComputationPlacer_NewFn(); in TpuComputationPlacer()
34 tensorflow::tpu::ExecutorApiFn()->TpuComputationPlacer_FreeFn(placer_); in ~TpuComputationPlacer()
47 tensorflow::tpu::ExecutorApiFn()->TpuComputationPlacer_AssignDevicesFn( in AssignDevices()
62 tensorflow::tpu::ExecutorApiFn()->TpuComputationPlacer_AssignLocalDevicesFn( in AssignLocalDevices()
|
| /external/tensorflow/tensorflow/core/tpu/kernels/xla/ |
| D | infeed_op.cc | 40 tpu::ExecutorApiFn()->TpuTransferManager_GetInfeedLayoutFn(&c_shape, in GetTPUInfeedLayout()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_tpu_device.cc | 62 tpu::ExecutorApiFn()->XlaShapeToTpuShapeRepresentationFn( in TpuShapeRepresentation()
93 tpu::ExecutorApiFn()->XlaShapeToTpuPaddedShapeFn( in TpuPaddedShapeFn()
|
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | infeed_ops.cc | 63 tpu::ExecutorApiFn()->TpuTransferManager_GetInfeedLayoutFn(&c_shape, in GetTPUInfeedLayout()
|