Home
last modified time | relevance | path

Searched defs:ExpandDim (Results 1 – 4 of 4) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/minddata/dataset/core/
Dcv_tensor.cc139 Status CVTensor::ExpandDim(const dsize_t &axis) { in ExpandDim() function in mindspore::dataset::CVTensor
Dtensor.cc738 Status Tensor::ExpandDim(const dsize_t &axis) { in ExpandDim() function in mindspore::dataset::Tensor
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/kernel_builder/ascend/akg/
Dcustom.py561 class ExpandDim(ExpandDimAndSqueeze): class
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/delegate/tensorrt/
Dtensorrt_utils.cc816 nvinfer1::ITensor *ExpandDim(TensorRTContext *ctx, nvinfer1::ITensor *input_tensor, int axis) { in ExpandDim() function