- Sort Score
- Result 10 results
- Languages All
Results 1 - 1 of 1 for FindDeviceFromOperands (0.13 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/tpu_device_propagation.cc
// devices, and empty StringRef will be returned. Control dependencies, // NextIteration.Source -> NextIteration.Sink token dependencies, and // LoopCond -> Switch data dependencies are ignored. llvm::StringRef FindDeviceFromOperands( Operation& op, const llvm::DenseMap<Value, llvm::StringRef>& value_to_device) { llvm::StringRef new_device; const bool is_switch = llvm::isa<tf_executor::SwitchOp>(op);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 11K bytes - Viewed (0)