Searched refs:ConfigureDistributedTpuOp (Results 1 – 2 of 2) sorted by relevance
39 class ConfigureDistributedTpuOp : public OpKernel {41 explicit ConfigureDistributedTpuOp(OpKernelConstruction* ctx) in ConfigureDistributedTpuOp() function48 ~ConfigureDistributedTpuOp() override {} in ~ConfigureDistributedTpuOp()52 ConfigureDistributedTpuOp(const ConfigureDistributedTpuOp&) = delete;53 ConfigureDistributedTpuOp& operator=(const ConfigureDistributedTpuOp&) =
110 void ConfigureDistributedTpuOp::Compute(OpKernelContext* ctx) { in Compute()410 ConfigureDistributedTpuOp);