Searched refs:TestRemoteFusedGraphExecutor (Results 1 – 4 of 4) sorted by relevance
96 TestRemoteFusedGraphExecutor::TestRemoteFusedGraphExecutor( in TestRemoteFusedGraphExecutor() function in tensorflow::TestRemoteFusedGraphExecutor101 int TestRemoteFusedGraphExecutor::GetVersion() { return 0; } in GetVersion()102 bool TestRemoteFusedGraphExecutor::Init(const RemoteFusedGraphExecuteInfo&) { in Init()105 bool TestRemoteFusedGraphExecutor::Finalize() { return true; } in Finalize()106 bool TestRemoteFusedGraphExecutor::SetupGraph() { return true; } in SetupGraph()107 bool TestRemoteFusedGraphExecutor::ExecuteGraph() { return true; } in ExecuteGraph()108 bool TestRemoteFusedGraphExecutor::TeardownGraph() { return true; } in TeardownGraph()109 bool TestRemoteFusedGraphExecutor::FillInputNode(const string&, const Tensor&) { in FillInputNode()112 bool TestRemoteFusedGraphExecutor::ReadOutputNode(const string&, in ReadOutputNode()116 Status TestRemoteFusedGraphExecutor::FuseRemoteGraph( in FuseRemoteGraph()[all …]
63 class TestRemoteFusedGraphExecutor final : public IRemoteFusedGraphExecutor {65 TestRemoteFusedGraphExecutor(const std::unordered_set<string>& fused_op_types,
61 new TestRemoteFusedGraphExecutor({"Mul"}, REMOTE_FUSED_EXECUTOR_NAME0)); in BuildRemoteFusedGraphExecutor0()67 executor->reset(new TestRemoteFusedGraphExecutor( in BuildRemoteFusedGraphExecutor1()
55 new TestRemoteFusedGraphExecutor({"Mul"}, REMOTE_FUSED_EXECUTOR_NAME0)); in BuildRemoteFusedGraphExecutor0()61 executor->reset(new TestRemoteFusedGraphExecutor( in BuildRemoteFusedGraphExecutor1()