Searched refs:TpuEmbedding_TensorBatchFixedState_Create_Params (Results 1 – 2 of 2) sorted by relevance
648 typedef struct TpuEmbedding_TensorBatchFixedState_Create_Params { struct657 } TpuEmbedding_TensorBatchFixedState_Create_Params; argument661 TpuEmbedding_TensorBatchFixedState_Create_Params* params);
84 TpuEmbedding_TensorBatchFixedState_Create_Params fixed_state_create_params; in EnqueueTPUEmbeddingArbitraryTensorBatchOp()