Function IsTypeSpecHelper

tensorflow/python/util/util.cc:506–514

Returns 1 if `o` is an instance of TypeSpec, but is not TensorSpec. Returns 0 otherwise. Returns -1 if an error occurred.

// Returns 0 otherwise.
// Returns -1 if an error occurred.
bool IsTypeSpecHelper(PyObject* o) {
  static auto* const check_cache = new CachedTypeCheck([](PyObject* to_check) {
    int is_type_spec = IsInstanceOfRegisteredType(to_check, "TypeSpec");
    int is_tensor_spec = IsInstanceOfRegisteredType(to_check, "TensorSpec");
    if ((is_type_spec == -1) || (is_tensor_spec == -1)) return -1;
    return static_cast<int>(is_type_spec && !is_tensor_spec);
  });
  return check_cache->CachedLookup(o);
}

// Returns 1 if `o` is a (non-string) sequence or CompositeTensor or
// (non-TensorSpec) TypeSpec.