onnxruntime
a021cb1b - Allow creating ConstantScalarNode for double type (#16797)

Commit
2 years ago
Allow creating ConstantScalarNode for double type (#16797) ### Allow creating ConstantScalarNode for double type Allow create ConstantScalarNode for double type. Looks double type is not respected when creating constant. So fix it. ``` onnxruntime::python::addObjectMethodsForTraining(pybind11::module&, onnxruntime::python::ExecutionProviderRegistrationFn)::<lambda(onnxruntime::training::OrtModuleGraphBuilder*, const onnxruntime::training::TrainingGraphTransformerConfiguration&)> [ONNXRuntimeError] : 1 : FAIL : Type Error: Type parameter (T) of Optype (Sub) bound to different types (tensor(double) and tensor(float) in node (/_original_module/_original_model/gpt_neox/layers.0/input_layernorm/Pow_Grad/Sub_1). ```
Author
Parents
Loading