Searched refs:GrapplerFunctionItem (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/utils/ |
| D | functions.h | 156 class GrapplerFunctionItem : public GrapplerItem {
158 GrapplerFunctionItem() = default;
179 GrapplerFunctionItem& SwapFunctionBody(GraphDef&& other);
184 GrapplerFunctionItem*);
186 GrapplerFunctionItem*);
188 GrapplerFunctionItem*,
191 GrapplerFunctionItem(string func_name, string description,
239 const GrapplerFunctionItem& item, const FunctionLibraryDefinition& flib,
244 GrapplerFunctionItem* item);
252 GrapplerFunctionItem* item,
[all …]
|
| D | functions.cc | 310 GrapplerFunctionItem::GrapplerFunctionItem( in GrapplerFunctionItem() function in tensorflow::grappler::GrapplerFunctionItem
348 const string& GrapplerFunctionItem::description() const { return description_; } in description()
350 const std::vector<InputArgExpansion>& GrapplerFunctionItem::inputs() const { in inputs()
354 const InputArgExpansion& GrapplerFunctionItem::input(int i) const { in input()
358 const std::size_t GrapplerFunctionItem::input_size() const { in input_size()
362 const std::vector<OutputArgExpansion>& GrapplerFunctionItem::outputs() const { in outputs()
366 const OutputArgExpansion& GrapplerFunctionItem::output(int i) const { in output()
370 const std::size_t GrapplerFunctionItem::output_size() const { in output_size()
374 const std::vector<ControlOutput>& GrapplerFunctionItem::control_outputs() in control_outputs()
379 const std::size_t GrapplerFunctionItem::control_output_size() const { in control_output_size()
[all …]
|
| D | functions_test.cc | 246 GrapplerFunctionItem item; in TEST_F()
328 GrapplerFunctionItem item; in TEST_F()
419 GrapplerFunctionItem item; in TEST_F()
482 GrapplerFunctionItem item; in TEST_F()
531 GrapplerFunctionItem item; in TEST_F()
596 GrapplerFunctionItem item; in TEST_F()
635 GrapplerFunctionItem item; in TEST_F()
662 GrapplerFunctionItem item; in TEST_F()
703 GrapplerFunctionItem item; in TEST_F()
743 GrapplerFunctionItem item; in TEST_F()
[all …]
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | function_optimizer.cc | 552 GrapplerFunctionItem* item, in PushDownConstInputs()
808 GrapplerFunctionItem item; in SpecializeFunction()
968 const GrapplerFunctionItem& item) { in InlinedFunctionInputsNode()
989 const NodeDef& func_node, const GrapplerFunctionItem& item, in InlinedFunctionOutputsNode()
1020 GrapplerFunctionItem item; in InlineDirectFunctionCall()
1309 const GrapplerFunctionItem& item, in MaybeDeadOutputs()
1327 GrapplerFunctionItem func_item; in MaybeDeadOutputs()
1472 const NodeDef& func_node, const GrapplerFunctionItem& item, in PlaceInlinedFunctionBody()
1570 GrapplerFunctionItem item; in InlineIndirectFunctionCall()
|
| D | meta_optimizer.cc | 598 GrapplerFunctionItem func_item; in Optimize()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
| D | rebatch.cc | 207 GrapplerFunctionItem f_item; in RecursivelyHandleOp()
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | graph_properties.cc | 609 GrapplerFunctionItem grappler_function_item = it->second; in UpdateFunction()
1044 GrapplerFunctionItem grappler_function_item; in AddFunction()
1694 std::unordered_map<string, GrapplerFunctionItem>
|