Searched refs:GraphInfo (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/lite/ |
| D | graph_info.h | 28 class GraphInfo {
30 virtual ~GraphInfo() {} in ~GraphInfo()
90 const GraphInfo* info, const TfLiteIntArray* nodes_to_partition,
|
| D | graph_info.cc | 40 const GraphInfo* info, const TfLiteIntArray* nodes_to_partition, in PartitionGraphIntoIndependentNodeSubsetsImpl()
198 const GraphInfo* info_;
218 const GraphInfo* info, const TfLiteIntArray* nodes_to_partition, in PartitionGraphIntoIndependentNodeSubsets()
|
| D | arena_planner.h | 52 ArenaPlanner(TfLiteContext* context, std::unique_ptr<GraphInfo> graph_info,
102 std::unique_ptr<GraphInfo> graph_info_;
|
| D | arena_planner_test.cc | 139 class TestGraphInfo : public GraphInfo {
182 &context_, std::unique_ptr<GraphInfo>(new TestGraphInfo(graph)), in SetGraph()
|
| D | arena_planner.cc | 38 std::unique_ptr<GraphInfo> graph_info, in ArenaPlanner()
|
| D | graph_info_test.cc | 37 class SimpleTestGraph : public GraphInfo {
|
| /external/tensorflow/tensorflow/core/graph/ |
| D | graph_partition.cc | 110 struct GraphInfo { struct
126 bool NeedSameDeviceSendRecv(const Edge* edge, const GraphInfo& info) { in NeedSameDeviceSendRecv()
148 bool IsDstInputOnHost(const Edge* edge, const GraphInfo& info) { in IsDstInputOnHost()
196 NodeDef* AddSend(const PartitionOptions& opts, const GraphInfo& g_info, in AddSend()
254 NodeDef* AddRecv(const PartitionOptions& opts, const GraphInfo& g_info, in AddRecv()
576 Status BuildMemoryDeviceInfo(const Graph& g, GraphInfo* info) { in BuildMemoryDeviceInfo()
636 GraphInfo* g_info) { in AddControlFlow()
985 GraphInfo g_info; in Partition()
|
| /external/tensorflow/tensorflow/lite/core/ |
| D | subgraph.cc | 176 class InterpreterInfo : public GraphInfo {
937 &context_, std::unique_ptr<GraphInfo>(new InterpreterInfo(this)), in PrepareOpsAndTensors()
|