- Sort Score
- Result 10 results
- Languages All
Results 1 - 1 of 1 for LoadTPUEmbeddingADAMParameters (0.44 sec)
-
tensorflow/compiler/mlir/tensorflow/ir/tf_generated_ops.td
); TF_DerivedOperandTypeAttr T = TF_DerivedOperandTypeAttr<0>; TF_DerivedResultTypeAttr out_idx = TF_DerivedResultTypeAttr<1>; } def TF_LoadTPUEmbeddingADAMParametersOp : TF_Op<"LoadTPUEmbeddingADAMParameters", [TF_MustExecute, TF_TPUEmbeddingReadEffect]> { let summary = "Load ADAM embedding parameters."; let description = [{ An op that loads optimization parameters into HBM for embedding. Must be
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 23:24:08 UTC 2024 - 793K bytes - Viewed (0)