Searched refs:GraphTransferer (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/hexagon/ |
| D | graph_transferer.cc | 78 GraphTransferer::GraphTransferer() { in GraphTransferer() function in tensorflow::GraphTransferer
82 GraphTransferer::~GraphTransferer() { delete graph_transfer_info_; } in ~GraphTransferer()
91 Status GraphTransferer::LoadGraphFromProto( in LoadGraphFromProto()
198 Status GraphTransferer::LoadGraphFromProtoFile( in LoadGraphFromProtoFile()
240 void GraphTransferer::SortParams(const std::vector<string>& output_node_names) { in SortParams()
283 void GraphTransferer::EnableStrictCheckMode(const bool enable) { in EnableStrictCheckMode()
287 void GraphTransferer::SetSerializedGraphTransferInfo( in SetSerializedGraphTransferInfo()
292 const GraphTransferInfo& GraphTransferer::GetGraphTransferInfo() const { in GetGraphTransferInfo()
296 GraphTransferInfo& GraphTransferer::GetMutableGraphTransferInfo() { in GetMutableGraphTransferInfo()
300 void GraphTransferer::CacheNode(const Node& node) { in CacheNode()
[all …]
|
| D | graph_transferer_test.cc | 47 GraphTransferer gt_;
195 const GraphTransferer& gt, const string& name) { in FindConstNodeInfo()
205 static const GraphTransferNodeInfo* FindNodeInfo(const GraphTransferer& gt, in FindNodeInfo()
217 const GraphTransferer& gt, const int node_id) { in FindNodeInputInfo()
228 const GraphTransferer& gt, const int node_id) { in FindNodeOutputInfo()
238 static void SanityCheckNodes(const GraphTransferer& gt) { in SanityCheckNodes()
372 TEST(GraphTransferer, LoadGraphFromProtoFile) { in TEST() argument
390 GraphTransferer gt; in TEST()
435 TEST(GraphTransferer, LoadGraphFromProtoFileShapeInferenceSimple) { in TEST() argument
454 GraphTransferer si_gt; in TEST()
[all …]
|
| D | hexagon_graph_execution_test.cc | 243 static void RunInferenceByHexagonControlWrapper(const GraphTransferer& gt, in RunInferenceByHexagonControlWrapper()
404 TEST(GraphTransferer, in TEST() argument
415 GraphTransferer gt; in TEST()
436 TEST(GraphTransferer, in TEST() argument
448 GraphTransferer gt; in TEST()
471 TEST(GraphTransferer, in TEST() argument
482 GraphTransferer gt; in TEST()
503 TEST(GraphTransferer, RunInceptionV3OnHexagonExampleWithTfRuntime) { in TEST() argument
531 TEST(GraphTransferer, DISABLED_RunInceptionV3OnHexagonExampleWithFusedGraph) { in TEST() argument
541 TEST(GraphTransferer, DISABLED_CheckShapeInferencePerformance) { in TEST() argument
[all …]
|
| D | graph_transferer.h | 44 class GraphTransferer {
52 GraphTransferer();
54 ~GraphTransferer();
226 TF_DISALLOW_COPY_AND_ASSIGN(GraphTransferer);
|
| D | hexagon_control_wrapper.h | 64 const std::array<int64, GraphTransferer::SHAPE_ARRAY_SIZE>& shape,
74 GraphTransferer graph_transferer_{};
|
| D | hexagon_control_wrapper.cc | 186 GraphTransferer gt; in SetupGraph()
323 const std::array<int64, GraphTransferer::SHAPE_ARRAY_SIZE>& shape, in FillInputNode()
430 const std::array<int64, GraphTransferer::SHAPE_ARRAY_SIZE> shape = in FillInputNode()
431 GraphTransferer::ToTensorShapeArray(tensor.shape()); in FillInputNode()
|