Searched refs:allow_mutable_tensors (Results 1 – 4 of 4) sorted by relevance
104 explicit FreezeGlobalTensorsPass(bool allow_mutable_tensors) { in FreezeGlobalTensorsPass()105 this->allow_mutable_tensors = allow_mutable_tensors; in FreezeGlobalTensorsPass()130 if (allow_mutable_tensors) continue; in runOnOperation()166 if (!allow_mutable_tensors) { in runOnOperation()232 if (!allow_mutable_tensors && !remaining_global_tensor_ops.empty()) { in runOnOperation()241 bool allow_mutable_tensors) { in CreateFreezeGlobalTensorsPass() argument242 return std::make_unique<FreezeGlobalTensorsPass>(allow_mutable_tensors); in CreateFreezeGlobalTensorsPass()
34 bool allow_mutable_tensors = false);
42 Option<"allow_mutable_tensors", "allow-mutable-tensors", "bool", /*default=*/"false",
5 // Test case: Do not fail if the tensor is mutable but allow_mutable_tensors is true