Home
last modified time | relevance | path

Searched defs:expand_dim (Results 1 – 3 of 3) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/delegate/tensorrt/op/
Dfullyconnected_tensorrt.cc98 std::vector<int64_t> expand_dim(origin_dims.d, origin_dims.d + origin_dims.nbDims); in PreprocessInputs() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/cuda_impl/cuda_class/
Dpad_v3_grad_helper.h72 size_t expand_dim = kMaxDim - input_shapes[0].size(); in CalMemSize() local
Dpad_v3_helper.h79 size_t expand_dim = kMaxDim - input_shapes[0].size(); in CalMemSize() local