limitations under the License.
==============================================================================*/
#ifndef TENSORFLOW_C_EAGER_C_API_REMOTE_TEST_UTIL_H_
#define TENSORFLOW_C_EAGER_C_API_REMOTE_TEST_UTIL_H_

// Run a function containing a MatMul op and check its output.
// If heavy_load_on_streaming_rpc is true, send some rpc requests before the one

#include "xla/pjrt/pjrt_c_api_client.h"
#include "xla/pjrt/tfrt_cpu_pjrt_client.h"
#include "xla/shape.h"
#include "xla/shape_util.h"
#include "tensorflow/core/framework/types.h"
#include "tensorflow/core/tfrt/common/async_value_tensor.h"
#include "tensorflow/core/tfrt/common/pjrt_util.h"
#include "tsl/lib/core/status_test_util.h"
#include "tsl/platform/casts.h"
#include "tsl/platform/status_matchers.h"
#include "tsl/protobuf/error_codes.pb.h"

#include "tensorflow/c/experimental/next_pluggable_device/tensor_pjrt_buffer_util.h"

#include <memory>
#include <utility>

#include "absl/status/status.h"
#include "absl/status/statusor.h"
#include "absl/strings/str_cat.h"
#include "tensorflow/compiler/jit/pjrt_tensor_buffer_util.h"
#include "xla/pjrt/c/pjrt_c_api.h"
#include "xla/pjrt/pjrt_c_api_client.h"
#include "xla/pjrt/pjrt_client.h"

See the License for the specific language governing permissions and
limitations under the License.
==============================================================================*/

#include "tensorflow/c/eager/c_api_test_util.h"

#include <memory>
#include <string>
#include <vector>

#include "tensorflow/c/eager/c_api.h"
#include "tensorflow/c/eager/c_api_experimental.h"
#include "tensorflow/c/eager/c_api_internal.h"

limitations under the License.
==============================================================================*/

#include "tensorflow/c/eager/c_api_remote_test_util.h"

#include "absl/strings/str_cat.h"
#include "tensorflow/c/eager/c_api_internal.h"
#include "tensorflow/c/eager/c_api_test_util.h"
#include "tensorflow/c/eager/tfe_tensorhandle_internal.h"
#include "tensorflow/core/common_runtime/eager/tensor_handle.h"

public static java.util.Map mergeMaps(java.util.Map, java.util.Map); public static java.util.Map mergeMaps(java.util.Map[]); public static java.util.Collection intersection(java.util.Collection, java.util.Collection); public static java.util.Collection subtract(java.util.Collection, java.util.Collection); public static java.util.Map getCardinalityMap(java.util.Collection); public static java.util.List iteratorToList(java.util.Iterator); private static final int getFreq(Object, java.util.Map); } org/c...

 * specific language governing permissions and limitations
 * under the License.
 */
package org.apache.maven.internal.impl;

import java.util.Collection;
import java.util.List;
import java.util.function.Function;
import java.util.stream.Collectors;

class Utils {
    static <T> T nonNull(T t) {
        if (t == null) {
            throw new IllegalArgumentException();
        }
        return t;
    }

public static java.util.Map mergeMaps(java.util.Map, java.util.Map); public static java.util.Map mergeMaps(java.util.Map[]); public static java.util.Collection intersection(java.util.Collection, java.util.Collection); public static java.util.Collection subtract(java.util.Collection, java.util.Collection); public static java.util.Map getCardinalityMap(java.util.Collection); public static java.util.List iteratorToList(java.util.Iterator); private static final int getFreq(Object, java.util.Map); } org/c...

public static java.util.Map mergeMaps(java.util.Map, java.util.Map); public static java.util.Map mergeMaps(java.util.Map[]); public static java.util.Collection intersection(java.util.Collection, java.util.Collection); public static java.util.Collection subtract(java.util.Collection, java.util.Collection); public static java.util.Map getCardinalityMap(java.util.Collection); public static java.util.List iteratorToList(java.util.Iterator); private static final int getFreq(Object, java.util.Map); } org/c...

 * specific language governing permissions and limitations
 * under the License.
 */
package org.apache.maven.internal.impl;

import java.util.Collection;
import java.util.List;
import java.util.function.Function;
import java.util.stream.Collectors;

class Utils {
    static <T> T nonNull(T t) {
        if (t == null) {
            throw new IllegalArgumentException();
        }
        return t;
    }