Searched refs:FunctionApiInfo (Results 1 – 3 of 3) sorted by relevance
26 FunctionApiInfo::FunctionApiInfo() {} in FunctionApiInfo() function in tensorflow::grappler::FunctionApiInfo27 FunctionApiInfo::~FunctionApiInfo() {} in ~FunctionApiInfo()29 Status FunctionApiInfo::Init(const FunctionDef& function_def) { in Init()30 function_type_ = FunctionApiInfo::FunctionType::INFERENCE; in Init()39 function_type_ = FunctionApiInfo::FunctionType::BACKWARD; in Init()43 function_type_ = FunctionApiInfo::FunctionType::FORWARD; in Init()65 const string& FunctionApiInfo::preferred_device() const { in preferred_device()69 const string& FunctionApiInfo::interface_name() const { in interface_name()73 const FunctionApiInfo::FunctionType FunctionApiInfo::function_type() const { in function_type()77 const string& FunctionApiInfo::pairing_function_name() const { in pairing_function_name()[all …]
31 class FunctionApiInfo {33 FunctionApiInfo();34 virtual ~FunctionApiInfo();63 TF_DISALLOW_COPY_AND_ASSIGN(FunctionApiInfo);82 const FunctionApiInfo* GetApiInfo(const string& function_name) const;88 std::unordered_map<string, std::unique_ptr<FunctionApiInfo>> func_info_;
148 const FunctionApiInfo& apiInfo) { in UpdateNodeDef()171 if (apiInfo.function_type() == FunctionApiInfo::BACKWARD) { in UpdateNodeDef()221 } else if (apiInfo.function_type() == FunctionApiInfo::FORWARD) { in UpdateNodeDef()