Home
last modified time | relevance | path

Searched defs:type_caster (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/python/util/
Dpy_checkpoint_reader_wrapper.cc55 struct type_caster<tensorflow::TensorShape> { struct
59 static handle cast(const tensorflow::TensorShape& src, in cast()
80 struct type_caster<tensorflow::DataType> { struct
84 static handle cast(const tensorflow::DataType& src, in cast()
/external/pytorch/torch/csrc/monitor/
Dpython_init.cpp21 struct type_caster<torch::monitor::data_value_t> { struct
26 bool load(handle src, bool) { in load()
43 static handle cast( in cast()
/external/federated-compute/fcp/tensorflow/python/
Dserve_slices_registry.cc39 struct type_caster<tensorflow::Tensor> { struct
42 static handle cast(const tensorflow::Tensor& tensor, return_value_policy, in cast()
/external/pytorch/torch/csrc/profiler/python/
Dinit.cpp85 struct type_caster<std::shared_ptr<torch::CapturedTraceback>> { struct
91 bool load(handle src, bool) { in load()
99 static handle cast( in cast()
/external/tensorflow/tensorflow/python/client/
Dtf_session_wrapper.cc52 struct type_caster<absl::optional<T>> struct
55 struct type_caster<absl::nullopt_t> : public void_caster<absl::nullopt_t> {}; struct
/external/pytorch/torch/csrc/jit/tensorexpr/
Dtensorexpr_init.cpp22 struct pybind11::detail::type_caster<torch::jit::tensorexpr::ArgValue> struct in pybind11::detail