to CamelCase

AndrewZhaoLuo · Sep 21, 2021 · d961c23 · d961c23
1 parent d3bc48f
commit d961c23
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/include/tvm/te/tensor.h b/include/tvm/te/tensor.h
@@ -107,7 +107,7 @@ class Tensor : public DataProducer {
    * \param support_negative_indices Whether to normalize indices in the case of negative indices.
    * \return the result expression representing tensor read.
    */
-  inline PrimExpr index_tensor(Array<PrimExpr> indices, bool support_negative_indices) const;
+  inline PrimExpr IndexTensor(Array<PrimExpr> indices, bool support_negative_indices) const;
 
  public:
   TVM_DLL Tensor(Array<PrimExpr> shape, DataType dtype, Operation op, int value_index);

diff --git a/src/te/tensor.cc b/src/te/tensor.cc
@@ -39,7 +39,7 @@ IterVar reduce_axis(Range dom, std::string name) { return IterVar(dom, Var(name)
 Var var(std::string name_hint, DataType t) { return Var(name_hint, t); }
 
 // Tensor
-inline PrimExpr Tensor::index_tensor(Array<PrimExpr> indices, bool support_negative_indices) const {
+inline PrimExpr Tensor::IndexTensor(Array<PrimExpr> indices, bool support_negative_indices) const {
   Array<PrimExpr> shape = (*this)->shape;
 
   if (shape.size() != 0) {
@@ -63,15 +63,15 @@ PrimExpr Tensor::operator()(Array<Var> indices) const {
   return operator()(arr);
 }
 
-PrimExpr Tensor::operator()(Array<PrimExpr> indices) const { return index_tensor(indices, false); }
+PrimExpr Tensor::operator()(Array<PrimExpr> indices) const { return IndexTensor(indices, false); }
 
 PrimExpr Tensor::IndexWithNegativeIndices(Array<Var> indices) const {
   Array<PrimExpr> arr(indices.begin(), indices.end());
   return IndexWithNegativeIndices(arr);
 }
 
 PrimExpr Tensor::IndexWithNegativeIndices(Array<PrimExpr> indices) const {
-  return index_tensor(indices, true);
+  return IndexTensor(indices, true);
 }
 
 String TensorNode::GetNameHint() const {