We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
2 parents 7cc6716 + dd6ebc4 commit eb136c9Copy full SHA for eb136c9
src/graph.rs
@@ -966,10 +966,14 @@ impl<'a> OperationDescription<'a> {
966
}
967
968
/// Sets an attribute which holds an array of tensors.
969
- pub fn set_attr_tensor_list<T: IntoIterator<Item = Tensor<u8>>>(&mut self,
970
- attr_name: &str,
971
- value: T)
972
- -> Result<()> {
+ pub fn set_attr_tensor_list<I, T>(
+ &mut self,
+ attr_name: &str,
+ value: I
973
+ ) -> Result<()>
974
+ where I: IntoIterator<Item = Tensor<T>>,
975
+ T: TensorType
976
+ {
977
let c_attr_name = CString::new(attr_name)?;
978
let mut status = Status::new();
979
unsafe {
0 commit comments