# tensorflow/core/api_def/base_api/*
# {op_name: {API's attribute name: OpDef's attribute name}}
_ATTRIBUTE_RENAMES = {
    'Mean': {'axis': 'reduction_indices'},
    'Split': {'axis': 'split_dim'},
    'SplitV': {'axis': 'split_dim'},
}


def _get_type_from_proto(arg_def=None, attr_def=None):
  if not arg_def:
    if attr_def.type == 'bool':
      return TFRTypes.I1
    elif attr_def.type == 'int32':

              Eq(BuiltinOperator_SPLIT));
  EXPECT_THAT(GetBuiltinCode(model_.operator_codes[add->opcode_index].get()),
              Eq(BuiltinOperator_ADD));

  // There should be 5 tensors: input, output, split, split/split_dim, split:1.
  // Tensor indices could be different between original and quantized.
  EXPECT_THAT(subgraph->tensors, SizeIs(5));
  const int input_idx = 0;
  EXPECT_THAT(subgraph->tensors[input_idx]->type, Eq(TensorType_INT8));