1 /* Copyright 2020 The TensorFlow Authors. All Rights Reserved. 2 3 Licensed under the Apache License, Version 2.0 (the "License"); 4 you may not use this file except in compliance with the License. 5 You may obtain a copy of the License at 6 7 http://www.apache.org/licenses/LICENSE-2.0 8 9 Unless required by applicable law or agreed to in writing, software 10 distributed under the License is distributed on an "AS IS" BASIS, 11 WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 12 See the License for the specific language governing permissions and 13 limitations under the License. 14 ==============================================================================*/ 15 #ifndef TENSORFLOW_C_EAGER_C_API_REMOTE_TEST_UTIL_H_ 16 #define TENSORFLOW_C_EAGER_C_API_REMOTE_TEST_UTIL_H_ 17 18 // Run a function containing a MatMul op and check its output. 19 // If heavy_load_on_streaming_rpc is true, send some rpc requests before the one 20 // which creates a remote input, to simulate a scenario that the remote input 21 // is not ready when we start running an op or a function. 22 void TestRemoteExecuteSilentCopies(bool async, bool remote, bool func, 23 bool heavy_load_on_streaming_rpc, 24 bool remote_func_outputs = false, 25 bool has_packed_input = false); 26 27 #endif // TENSORFLOW_C_EAGER_C_API_REMOTE_TEST_UTIL_H_ 28