Searched refs:SE_StreamExecutor (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/tpu/ |
| D | tpu_executor_c_api.h | 34 SE_StreamExecutor* TpuPlatform_GetExecutor(SE_Platform* platform,
45 void TpuExecutor_Init(SE_StreamExecutor* executor, int device_ordinal,
47 void TpuExecutor_Free(SE_StreamExecutor* executor);
49 int TpuExecutor_PlatformDeviceCount(SE_StreamExecutor* executor);
51 SE_DeviceMemoryBase TpuExecutor_Allocate(SE_StreamExecutor* executor,
53 void TpuExecutor_Deallocate(SE_StreamExecutor* executor,
55 bool TpuExecutor_GetAllocatorStats(SE_StreamExecutor* executor,
57 bool TpuExecutor_DeviceMemoryUsage(SE_StreamExecutor* executor, int64_t* free,
60 bool TpuExecutor_AllocateStream(SE_StreamExecutor* executor, SE_Stream* stream);
61 void TpuExecutor_DeallocateStream(SE_StreamExecutor* executor,
[all …]
|
| D | c_api_decl.h | 54 typedef struct SE_StreamExecutor SE_StreamExecutor; typedef
292 SE_StreamExecutor** exec;
|
| D | tpu_executor.h | 61 SE_StreamExecutor* executor) in TpuExecutor()
220 SE_StreamExecutor* se_executor() { return executor_; } in se_executor()
238 SE_StreamExecutor* executor_;
|
| D | c_api_defn.h | 38 struct SE_StreamExecutor { struct
|
| D | tpu_platform.cc | 97 SE_StreamExecutor* executor = tpu::ExecutorApiFn()->TpuPlatform_GetExecutorFn( in GetUncachedExecutor()
|
| D | tpu_transfer_manager.cc | 153 std::vector<SE_StreamExecutor*> se; in ResetDevices()
|
| /external/tensorflow/tensorflow/core/tpu/ |
| D | tpu_on_demand_compiler.cc | 297 std::vector<std::vector<SE_StreamExecutor*>> se_lists_storage; in Compile()
|