Skip to content
This repository was archived by the owner on Nov 17, 2023. It is now read-only.
This repository was archived by the owner on Nov 17, 2023. It is now read-only.

Ensure all usage of ndim to be 0-d tensor compatible #14342

@junrushao

Description

@junrushao

EDIT: See also #14323.

After #14315, we should check all the occurrence of ndim to ensure they are used correctly in the semantics of 0-d tensor.

@reminisce @szha @eric-haibin-lin @zheng-da @yzhliu

List of occurrence

src/c_api

  • src/c_api/c_api.cc
  • src/c_api/c_api_common.h
  • src/c_api/c_api_symbolic.cc
  • src/c_api/c_predict_api.cc

src/common

  • src/common/serialization.h
  • src/common/static_array.h
  • src/common/utils.h

src/executor

  • src/executor/infer_graph_attr_pass.cc

src/imperative

  • src/imperative/cached_op.cc
  • src/imperative/imperative_utils.h

src/io

  • src/io/image_aug_default.cc
  • src/io/image_det_aug_default.cc
  • src/io/image_io.cc
  • src/io/image_iter_common.h
  • src/io/iter_batchloader.h
  • src/io/iter_image_det_recordio.cc
  • src/io/iter_image_recordio_2.cc
  • src/io/iter_libsvm.cc
  • src/io/iter_sparse_batchloader.h

src/kvstore

  • src/kvstore/gradient_compression.cc
  • src/kvstore/kvstore_dist.h
  • src/kvstore/kvstore_dist_server.h

src/ndarray

  • src/ndarray/ndarray.cc
  • src/ndarray/ndarray_function.cc
  • src/ndarray/ndarray_function.cu
  • src/ndarray/ndarray_function.h

src/nnvm

Nothing here.

include

  • include/mxnet/ndarray.h
  • include/mxnet/operator.h
  • include/mxnet/operator_util.h
  • include/mxnet/resource.h
  • include/mxnet/tensor_blob.h
  • include/mxnet/tuple.h

Metadata

Metadata

Assignees

No one assigned

    Labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions