- Sort Score
- Result 10 results
- Languages All
Results 1 - 1 of 1 for TPUEmbeddingOpsRegistry (0.24 sec)
-
tensorflow/compiler/mlir/tf2xla/transforms/legalization_op_config.cc
TypeID::get<TF::XlaSpmdShardToFullShapeOp>(), TypeID::get<TF::XlaSvdOp>(), }; // Add the ops from the TPUEmbeddingOpsRegistry. for (auto op_type_id : TF::TPUEmbeddingOpsRegistry::Global().GetOpsTypeIds()) { ops_set->insert(op_type_id); } return ops_set; }(); return ops->count(type_id); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 24 04:08:35 UTC 2024 - 21.7K bytes - Viewed (0)