Searched refs:GrapplerFunctionItem (Results 1 – 8 of 8) sorted by relevance
61 class GrapplerFunctionItem : public GrapplerItem {63 GrapplerFunctionItem() = default;85 GrapplerFunctionItem& SwapFunctionBody(GraphDef&& other);90 GrapplerFunctionItem*);92 GrapplerFunctionItem*);94 GrapplerFunctionItem*,97 GrapplerFunctionItem(string func_name, string description,147 GrapplerFunctionItem* item);153 GrapplerFunctionItem* item,165 GrapplerFunctionItem* item);[all …]
39 GrapplerFunctionItem::GrapplerFunctionItem( in GrapplerFunctionItem() function in tensorflow::grappler::GrapplerFunctionItem75 const string& GrapplerFunctionItem::description() const { return description_; } in description()77 const std::vector<InputArgInstantiation>& GrapplerFunctionItem::inputs() const { in inputs()81 const InputArgInstantiation& GrapplerFunctionItem::input(int i) const { in input()85 const std::size_t GrapplerFunctionItem::input_size() const { in input_size()89 const std::vector<OutputArgInstantiation>& GrapplerFunctionItem::outputs() in outputs()94 const OutputArgInstantiation& GrapplerFunctionItem::output(int i) const { in output()98 const std::size_t GrapplerFunctionItem::output_size() const { in output_size()102 const std::vector<ControlOutput>& GrapplerFunctionItem::control_outputs() in control_outputs()107 const std::size_t GrapplerFunctionItem::control_output_size() const { in control_output_size()[all …]
130 GrapplerFunctionItem item; in TEST_F()211 GrapplerFunctionItem item; in TEST_F()306 GrapplerFunctionItem item; in TEST_F()371 GrapplerFunctionItem item; in TEST_F()423 GrapplerFunctionItem item; in TEST_F()462 GrapplerFunctionItem item; in TEST_F()489 GrapplerFunctionItem item; in TEST_F()538 GrapplerFunctionItem item; in TEST_F()587 GrapplerFunctionItem item; in TEST_F()681 GrapplerFunctionItem item; in TEST_F()[all …]
127 GrapplerFunctionItem func_item; in Optimize()
113 GrapplerFunctionItem item; in TEST()
826 GrapplerFunctionItem func_item; in OptimizeConsumeItem()876 GrapplerFunctionItem func_item_copy = func_item; in OptimizeConsumeItem()
464 GrapplerFunctionItem* item, in PushDownConstInputs()718 GrapplerFunctionItem item; in SpecializeFunction()
766 const absl::optional<GrapplerFunctionItem>& maybe_grappler_function_item = in UpdateFunction()784 GrapplerFunctionItem grappler_function_item = *maybe_grappler_function_item; in UpdateFunction()1277 GrapplerFunctionItem grappler_function_item; in AddFunction()2006 absl::flat_hash_map<string, absl::optional<GrapplerFunctionItem>>