- Sort Score
- Result 10 results
- Languages All
Results 1 - 1 of 1 for SupportsCommunicationComputation (0.18 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/prepare_tpu_computation_for_tf_export.cc
op); } // Returns true if the given op is one of ops supported to have communication // subcomputation in the TF/XLA bridge. bool SupportsCommunicationComputation(Operation* op) { return isa<TF::IfRegionOp, TF::WhileRegionOp, TF::CaseRegionOp, TF::XlaCallModuleOp, TF::StatefulPartitionedCallOp, TF::PartitionedCallOp, TF::LegacyCallOp>(op); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 11.8K bytes - Viewed (0)