Home
last modified time | relevance | path

Searched refs:PythonAPIInfo (Results 1 – 9 of 9) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/python/framework/
H A Dpython_api_info_wrapper.cc28 void InitializeFromRegisteredOp(PythonAPIInfo* api_info, in InitializeFromRegisteredOp()
38 PythonAPIInfo* api_info, in InitializeFromParamSpecs()
50 std::string DebugInfo(PythonAPIInfo* api_info) { return api_info->DebugInfo(); } in DebugInfo()
55 using PythonAPIInfo = tensorflow::PythonAPIInfo; typedef
56 using InferredAttributes = tensorflow::PythonAPIInfo::InferredAttributes;
59 py::class_<PythonAPIInfo>(m, "PythonAPIInfo") in PYBIND11_MODULE()
66 [](PythonAPIInfo* self) { return self->inferred_type_attrs(); }) in PYBIND11_MODULE()
68 [](PythonAPIInfo* self) { return self->inferred_type_list_attrs(); }) in PYBIND11_MODULE()
70 [](PythonAPIInfo* self) { return self->inferred_length_attrs(); }); in PYBIND11_MODULE()
H A Dpython_api_parameter_converter.cc45 using InferredAttributes = PythonAPIInfo::InferredAttributes;
46 using ParamIndex = PythonAPIInfo::ParamIndex;
47 using Attribute = PythonAPIInfo::Attribute;
48 using InputWithFixedDType = PythonAPIInfo::InputWithFixedDType;
49 using InputsWithTypeAttr = PythonAPIInfo::InputsWithTypeAttr;
50 using InputsWithTypeListAttr = PythonAPIInfo::InputsWithTypeListAttr;
158 const PythonAPIInfo& api_info, int param_index, in ConvertToTensorInPlace()
198 bool ConvertAttribute(const Attribute& attr, const PythonAPIInfo& api_info, in ConvertAttribute()
221 const PythonAPIInfo& api_info, in ConvertInputWithFixedDType()
248 const PythonAPIInfo& api_info, in ConvertInputsWithTypeAttr()
[all …]
H A Dpython_api_info.cc118 PythonAPIInfo::PythonAPIInfo(const std::string& api_name) in PythonAPIInfo() function in tensorflow::PythonAPIInfo
121 Status PythonAPIInfo::Initialize(const OpDef& op_def, in Initialize()
173 Status PythonAPIInfo::CheckParamNames() const { in CheckParamNames()
196 Status PythonAPIInfo::InitializeFromRegisteredOp(const std::string& op_name) { in InitializeFromRegisteredOp()
207 Status PythonAPIInfo::InitializeFromParamSpecs( in InitializeFromParamSpecs()
229 Status PythonAPIInfo::InitializeAttribute( in InitializeAttribute()
299 Status PythonAPIInfo::InitializeInput( in InitializeInput()
366 PythonAPIInfo::InputsWithTypeAttr* PythonAPIInfo::FindInputsWithTypeAttr( in FindInputsWithTypeAttr()
376 PythonAPIInfo::InputsWithTypeListAttr*
377 PythonAPIInfo::FindInputsWithTypeListAttr(const string& name) { in FindInputsWithTypeListAttr()
[all …]
H A Dpython_api_parameter_converter.h56 const PythonAPIInfo& api_info,
58 PythonAPIInfo::InferredAttributes* inferred_attrs);
69 bool CopyPythonAPITensorLists(const PythonAPIInfo& api_info,
H A Dpython_api_parameter_converter_wrapper.cc28 PythonAPIInfo::InferredAttributes Convert( in Convert()
29 const PythonAPIInfo& api_info, in Convert()
31 PythonAPIInfo::InferredAttributes inferred_attrs; in Convert()
H A Dpython_api_info_test.py42 api_info = _pywrap_python_api_info.PythonAPIInfo(op_name)
53 api_info = _pywrap_python_api_info.PythonAPIInfo(api_name)
H A Dpython_api_info.h58 class PythonAPIInfo {
136 explicit PythonAPIInfo(const std::string& api_name);
H A Dpython_api_parameter_converter_test.py56 api_info = _pywrap_python_api_info.PythonAPIInfo(op_name)
67 api_info = _pywrap_python_api_info.PythonAPIInfo(api_name)
/aosp_15_r20/external/tensorflow/tensorflow/tools/def_file_filter/
H A Dsymbols_pybind.txt475 tensorflow::PythonAPIInfo