Home
last modified time | relevance | path

Searched refs:RemoteFusedGraphExecuteInfo (Results 1 – 16 of 16) 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_rewriter_transform_test.cc188 RemoteFusedGraphExecuteInfo info; in CheckGraph()
Dremote_fused_graph_execute_utils.cc551 const RemoteFusedGraphExecuteInfo& proto, in BuildRemoteGraphInputsAndOutputsFromProto()
610 const bool require_shape_type, RemoteFusedGraphExecuteInfo* execute_info, in BuildRemoteFusedGraphExecuteInfo()
630 RemoteFusedGraphExecuteInfo::TensorShapeTypeProto& tensor_shape_type = in BuildRemoteFusedGraphExecuteInfo()
654 RemoteFusedGraphExecuteInfo::TensorShapeTypeProto& in BuildRemoteFusedGraphExecuteInfo()
684 RemoteFusedGraphExecuteInfo execute_info; in BuildRemoteFusedGraphExecuteOpNode()
Dremote_fused_graph_execute_utils_test.cc438 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()
588 const RemoteFusedGraphExecuteInfo ei0 = in TEST()
590 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/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/core/framework/
Dremote_fused_graph_execute_info.proto18 message RemoteFusedGraphExecuteInfo { message