if (!type.hasRank())
    return std::make_pair(dtype, tensorflow::PartialTensorShape());

  auto shape = type.getShape();
  llvm::SmallVector<int64_t, 4> dims;
  dims.assign(shape.begin(), shape.end());
  return std::make_pair(dtype, tensorflow::PartialTensorShape(dims));
}

}  // namespace

Status MapFunctionSignaturesFromTFSavedModelMLIR(
    mlir::ModuleOp module,

    left_padding = split_outputs[0]
    core = split_outputs[1]
    right_padding = split_outputs[2]
    reversed_left_padding = tf.raw_ops.Reverse(tensor=left_padding, dims=rdims)
    reversed_right_padding = tf.raw_ops.Reverse(
        tensor=right_padding, dims=rdims)
    zero_like = tf.raw_ops.ZerosLike(x=core)
    left_offset, _ = tf.raw_ops.SplitV(
        value=zero_like,
        size_splits=[-1, left_padding_size],

reviewers:
  - thockin
  - smarterclayton
  - wojtek-t
  - deads2k
  - caesarxuchao
  - liggitt
  - sttts
  - luxas
  - janetkuo
  - justinsb
  - ncdc
  - soltysh

reviewers:
  - thockin
  - smarterclayton
  - wojtek-t
  - deads2k
  - derekwaynecarr
  - caesarxuchao
  - mikedanese
  - liggitt
  - saad-ali
  - janetkuo
  - tallclair
  - dims

reviewers:
  - thockin
  - smarterclayton
  - wojtek-t
  - deads2k
  - yujuhong
  - derekwaynecarr
  - caesarxuchao
  - mikedanese
  - liggitt
  - saad-ali
  - janetkuo
  - pwittrock
  - ncdc
  - dims

  - yujuhong
  - derekwaynecarr
  - caesarxuchao
  - mikedanese
  - liggitt
  - sttts
  - dchen1107
  - janetkuo
  - justinsb
  - pwittrock
  - tallclair
  - soltysh
  - jsafrane

  - wojtek-t
  - deads2k
  - derekwaynecarr
  - caesarxuchao
  - mikedanese
  - liggitt
  - sttts
  - saad-ali
  - janetkuo
  - justinsb
  - ncdc
  - tallclair
  - mwielgus
  - soltysh

// Then:
//  * D,E,F are the update window dims [2,3,4]
//  * C is the index vector dimension
//  * A,B iterate over the updates and indices
//
// If `update_window_dims` are not the trailing dimensions then updates must be
// transposed.
//
// Eg 2. An update in non-canonical form:
//  * indices shape(a,b,c)
//  * updates shape(d,e,f,a,b)
// Then:
//  * d,e,f are the update window dims [0,1,2]
//  * c is the index vector dimension

  - wojtek-t
  - deads2k
  - derekwaynecarr
  - caesarxuchao
  - mikedanese
  - liggitt
  - sttts
  - saad-ali
  - janetkuo
  - justinsb
  - ncdc
  - tallclair
  - mwielgus
  - soltysh
  - dims
  - jpbetz
labels:

  Status SetAttrBool(const char* attr_name, bool value) override;
  Status SetAttrType(const char* attr_name, DataType value) override;
  Status SetAttrShape(const char* attr_name, const int64_t* dims,
                      const int num_dims) override;
  Status SetAttrFunction(const char* attr_name,
                         const AbstractOperation* value) override;