Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/dtensor/cc/
H A Ddtensor_tpu_kernels.cc181 params.struct_size = InitializeHostForDistributedTpuOp_DoWork_Params_SIZE; in InitializeInternal()
/aosp_15_r20/external/tensorflow/tensorflow/core/tpu/kernels/
H A Dtpu_configuration_ops.cc331 params.struct_size = InitializeHostForDistributedTpuOp_DoWork_Params_SIZE; in Compute()
/aosp_15_r20/external/tensorflow/tensorflow/core/tpu/
H A Dtpu_ops_c_api.h326 #define InitializeHostForDistributedTpuOp_DoWork_Params_SIZE \ macro