Searched defs:DTensorShardedPrefixOpKernel (Results 1 – 1 of 1) sorted by relevance
30 class DTensorShardedPrefixOpKernel : public OpKernel { class32 explicit DTensorShardedPrefixOpKernel(OpKernelConstruction* ctx) in DTensorShardedPrefixOpKernel() function in tensorflow::dtensor::DTensorShardedPrefixOpKernel