#version 460 core
#extension GL_ARM_tensors : enable
#extension GL_EXT_shader_explicit_arithmetic_types : enable

layout(set = 0, binding = 0) uniform tensorARM<int32_t, 4> tu;
layout(set = 0, binding = 1) uniform tensorARM<int32_t, 2> td;

layout(set = 0, binding = 2, std430) buffer buff {
  uint out_data[];
};

void unusedTensorParam(tensorARM) {
  return;
}

uint getDim0Size(tensorARM t) {
  return tensorSizeARM(t, 0);
}

uint partiallySpecialized(tensorARM<int> t) {
  return tensorSizeARM(t, 1);
}

void main() {
  unusedTensorParam(tu);
  unusedTensorParam(td);
  out_data[0] = getDim0Size(tu);
  out_data[0] += getDim0Size(td);
  out_data[0] += partiallySpecialized(tu);
}