Home
last modified time | relevance | path

Searched refs:RemoteFusedGraphExecuteInfo (Results 1 – 18 of 18) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dremote_fused_graph_execute_op_test.cc105 const RemoteFusedGraphExecuteInfo& execute_info, Scope* root) { in BuildRemoteFusedGraphExecuteOp()
128 static RemoteFusedGraphExecuteInfo BuildRemoteFusedGraphExecuteInfo( in BuildRemoteFusedGraphExecuteInfo()
130 RemoteFusedGraphExecuteInfo execute_info; in BuildRemoteFusedGraphExecuteInfo()
147 RemoteFusedGraphExecuteInfo::TensorShapeTypeProto& shape_a = in BuildRemoteFusedGraphExecuteInfo()
154 RemoteFusedGraphExecuteInfo::TensorShapeTypeProto& shape_a_plus_b = in BuildRemoteFusedGraphExecuteInfo()
166 bool Init(const RemoteFusedGraphExecuteInfo& info) final { in Init()
228 const RemoteFusedGraphExecuteInfo* info_;
257 const RemoteFusedGraphExecuteInfo execute_info = in RewriteGraphToFusedGraph()
Di_remote_fused_graph_executor.h26 class RemoteFusedGraphExecuteInfo; variable
42 virtual bool Init(const RemoteFusedGraphExecuteInfo& info) = 0;
Dremote_fused_graph_execute_utils.h40 class RemoteFusedGraphExecuteInfo; variable
166 const RemoteFusedGraphExecuteInfo& proto,
178 const bool require_shape_type, RemoteFusedGraphExecuteInfo* execute_info,
Dremote_fused_graph_execute_op_test_utils.h69 bool Init(const RemoteFusedGraphExecuteInfo&) final;
Dremote_fused_graph_execute_op.cc126 RemoteFusedGraphExecuteInfo execute_info_;
Dremote_fused_graph_execute_op_test_utils.cc102 bool TestRemoteFusedGraphExecutor::Init(const RemoteFusedGraphExecuteInfo&) { in Init() argument
Dremote_fused_graph_execute_utils.cc550 const RemoteFusedGraphExecuteInfo& proto, in BuildRemoteGraphInputsAndOutputsFromProto()
609 const bool require_shape_type, RemoteFusedGraphExecuteInfo* execute_info, in BuildRemoteFusedGraphExecuteInfo()
629 RemoteFusedGraphExecuteInfo::TensorShapeTypeProto& tensor_shape_type = in BuildRemoteFusedGraphExecuteInfo()
653 RemoteFusedGraphExecuteInfo::TensorShapeTypeProto& in BuildRemoteFusedGraphExecuteInfo()
683 RemoteFusedGraphExecuteInfo execute_info; in BuildRemoteFusedGraphExecuteOpNode()
Dremote_fused_graph_rewriter_transform_test.cc188 RemoteFusedGraphExecuteInfo info; in CheckGraph()
Dremote_fused_graph_execute_utils_test.cc439 RemoteFusedGraphExecuteInfo execute_info0; in TEST()
/external/tensorflow/tensorflow/core/kernels/hexagon/
Dgraph_transfer_utils.cc58 /* static */ RemoteFusedGraphExecuteInfo
64 RemoteFusedGraphExecuteInfo execute_info; in BuildRemoteFusedGraphExecuteInfo()
72 RemoteFusedGraphExecuteInfo::TensorShapeTypeProto& tensor_shape_type = in BuildRemoteFusedGraphExecuteInfo()
87 RemoteFusedGraphExecuteInfo::TensorShapeTypeProto& tensor_shape_type_proto = in BuildRemoteFusedGraphExecuteInfo()
131 const RemoteFusedGraphExecuteInfo execute_info = in BuildFusedGraphDef()
Dgraph_transfer_utils.h29 class RemoteFusedGraphExecuteInfo; variable
48 static RemoteFusedGraphExecuteInfo BuildRemoteFusedGraphExecuteInfo(
Dhexagon_control_wrapper.h43 bool Init(const RemoteFusedGraphExecuteInfo& info) final;
73 const RemoteFusedGraphExecuteInfo* execute_info_{};
Dhexagon_graph_execution_test.cc209 /* static */ RemoteFusedGraphExecuteInfo
212 RemoteFusedGraphExecuteInfo execute_info; in BuildRemoteFusedGraphExecuteInfoWithGraphTransferInfo()
217 RemoteFusedGraphExecuteInfo::TensorShapeTypeProto& tensor_shape_type = in BuildRemoteFusedGraphExecuteInfoWithGraphTransferInfo()
229 RemoteFusedGraphExecuteInfo::TensorShapeTypeProto& tensor_shape_type = in BuildRemoteFusedGraphExecuteInfoWithGraphTransferInfo()
245 const RemoteFusedGraphExecuteInfo execute_info = in RunInferenceByHexagonControlWrapper()
590 const RemoteFusedGraphExecuteInfo ei0 = in TEST()
592 const RemoteFusedGraphExecuteInfo ei1 = in TEST()
Dhexagon_control_wrapper.cc74 bool HexagonControlWrapper::Init(const RemoteFusedGraphExecuteInfo& info) { in Init()
108 const RemoteFusedGraphExecuteInfo::TensorShapeTypeProto& shape_type = in Init()
119 const RemoteFusedGraphExecuteInfo::TensorShapeTypeProto& shape_type = in Init()
/external/tensorflow/tensorflow/contrib/hvx/hvx_ops_support_checker/
Dhvx_ops_support_checker_main.cc80 RemoteFusedGraphExecuteInfo info; in DumpRemoteFusedGraph()
93 for (const RemoteFusedGraphExecuteInfo::TensorShapeTypeProto& shape_type : in DumpRemoteFusedGraph()
100 for (const RemoteFusedGraphExecuteInfo::TensorShapeTypeProto& shape_type : in DumpRemoteFusedGraph()
/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_RemoteFusedGraphExecute.pbtxt19 of RemoteFusedGraphExecuteInfo which contains graph specifications.
25 are stored as a serialized protocol buffer of RemoteFusedGraphExecuteInfo
/external/tensorflow/tensorflow/contrib/remote_fused_graph/pylib/python/ops/
Dremote_fused_graph_ops.py44 info_proto = info_pb2.RemoteFusedGraphExecuteInfo()
/external/tensorflow/tensorflow/core/framework/
Dremote_fused_graph_execute_info.proto16 message RemoteFusedGraphExecuteInfo { message