Use TensorType's datatype instead of hardcoded DataType::Float
Use TensorType's datatype instead of hardcoded DataType::Float

adamcrume/tensorflow-rust,adamcrume/tensorflow-rust,adamcrume/tensorflow-rust,adamcrume/tensorflow-rust,tensorflow/rust,tensorflow/rust,tensorflow/rust,tensorflow/rust