tensorflow/tensorflow/tensorflow/dtensor/mlir/ir/ Source Tree

../
tf_dtensor.cc
tf_dtensor.cc.inc
tf_dtensor.h
tf_dtensor.h.inc ApplyEmbeddingOptimizerOp ApplyEmbeddingOptimizerOpAdaptor ApplyEmbeddingOptimizerV2Op ApplyEmbeddingOptimizerV2OpAdaptor CopyToMeshOp CopyToMeshOpAdaptor DTensorAllGatherOp DTensorAllGatherOpAdaptor DTensorAllReduceOp DTensorAllReduceOpAdaptor DTensorAllScatterOp DTensorAllScatterOpAdaptor DTensorEmbeddingDequeueOp DTensorEmbeddingDequeueOpAdaptor DTensorEmbeddingEnqueueOp DTensorEmbeddingEnqueueOpAdaptor DTensorLayout DTensorLayoutAdaptor DTensorLoadTPUEmbeddingParametersOp DTensorLoadTPUEmbeddingParametersOpAdaptor DTensorRecv DTensorRecvAdaptor DTensorReduceScatterOp DTensorReduceScatterOpAdaptor DTensorRestoreV2Op DTensorRestoreV2OpAdaptor DTensorRetrieveTPUEmbeddingParametersOp DTensorRetrieveTPUEmbeddingParametersOpAdaptor DTensorSend DTensorSendAdaptor DTensorSendEmbeddingGradientsOp DTensorSendEmbeddingGradientsOpAdaptor DTensorShardedPrefixOp DTensorShardedPrefixOpAdaptor EmbeddingDequeueOp EmbeddingDequeueOpAdaptor EmbeddingEnqueueOp EmbeddingEnqueueOpAdaptor GetEmbeddingConfigurationOp GetEmbeddingConfigurationOpAdaptor IsTPUEmbeddingInitializedOp IsTPUEmbeddingInitializedOpAdaptor LoadAllTPUEmbeddingParametersOp LoadAllTPUEmbeddingParametersOpAdaptor LoadEmbeddingTableOp LoadEmbeddingTableOpAdaptor RelayoutOp RelayoutOpAdaptor RetrieveAllTPUEmbeddingParametersOp RetrieveAllTPUEmbeddingParametersOpAdaptor SetEmbeddingConfigOp SetEmbeddingConfigOpAdaptor TPUDenseEmbeddingLookUpGradOp TPUDenseEmbeddingLookUpGradOpAdaptor TPUDenseEmbeddingLookUpOp TPUDenseEmbeddingLookUpOpAdaptor

Browse the source code of tensorflow/tensorflow/tensorflow/dtensor/mlir/ir/ online