Searched refs:GraphInfo (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/lite/ |
| D | graph_info.h | 26 class GraphInfo {
28 virtual ~GraphInfo() {} in ~GraphInfo()
78 const GraphInfo* info, const TfLiteIntArray* nodes_to_partition,
|
| D | graph_info.cc | 55 const GraphInfo* info, const TfLiteIntArray* nodes_to_partition, in PartitionGraphIntoIndependentNodeSubsetsImpl()
209 const GraphInfo* info_;
225 const GraphInfo* info, const TfLiteIntArray* nodes_to_partition, in PartitionGraphIntoIndependentNodeSubsets()
|
| D | arena_planner.h | 53 ArenaPlanner(TfLiteContext* context, std::unique_ptr<GraphInfo> graph_info,
95 std::unique_ptr<GraphInfo> graph_info_;
|
| D | arena_planner_test.cc | 128 class TestGraphInfo : public GraphInfo {
168 &context_, std::unique_ptr<GraphInfo>(new TestGraphInfo(graph)), in SetGraph()
|
| D | arena_planner.cc | 30 std::unique_ptr<GraphInfo> graph_info, in ArenaPlanner()
|
| D | graph_info_test.cc | 33 class SimpleTestGraph : public GraphInfo {
|
| /external/tensorflow/tensorflow/core/graph/ |
| D | graph_partition.cc | 108 struct GraphInfo { struct
124 bool NeedSameDeviceSendRecv(const Edge* edge, const GraphInfo& info) { in NeedSameDeviceSendRecv()
146 bool IsDstInputOnHost(const Edge* edge, const GraphInfo& info) { in IsDstInputOnHost()
192 NodeDef* AddSend(const PartitionOptions& opts, const GraphInfo& g_info, in AddSend()
250 NodeDef* AddRecv(const PartitionOptions& opts, const GraphInfo& g_info, in AddRecv()
549 Status BuildMemoryDeviceInfo(const Graph& g, GraphInfo* info) { in BuildMemoryDeviceInfo()
609 GraphInfo* g_info) { in AddControlFlow()
958 GraphInfo g_info; in Partition()
|
| /external/tensorflow/tensorflow/lite/core/ |
| D | subgraph.cc | 110 class InterpreterInfo : public GraphInfo {
641 context_, std::unique_ptr<GraphInfo>(new InterpreterInfo(this)), in PrepareOpsAndTensors()
|