Home
last modified time | relevance | path

Searched refs:InitializeHostForDistributedTpuOp_DoWork_Params (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/tpu/
H A Dtpu_ops_c_api.h312 typedef struct InitializeHostForDistributedTpuOp_DoWork_Params { struct
324 } InitializeHostForDistributedTpuOp_DoWork_Params; argument
327 (sizeof(struct InitializeHostForDistributedTpuOp_DoWork_Params))
330 InitializeHostForDistributedTpuOp_DoWork_Params* params);
/aosp_15_r20/external/tensorflow/tensorflow/dtensor/cc/
H A Ddtensor_tpu_kernels.cc180 InitializeHostForDistributedTpuOp_DoWork_Params params; in InitializeInternal()
/aosp_15_r20/external/tensorflow/tensorflow/core/tpu/kernels/
H A Dtpu_configuration_ops.cc330 InitializeHostForDistributedTpuOp_DoWork_Params params; in Compute()