Home
last modified time | relevance | path

Searched defs:shape_c (Results 1 – 5 of 5) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/auto_parallel/rec_core/
Drec_tensor.h29 int64_t shape_c = 1; member
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/math/
Deinsum_grad_gpu_kernel.h217 auto shape_c = Convert2SizeTClipNeg(std::get<IDX_OUT_SHAPE>(res_op_[idx_op])); in ResOpBackward() local
356 auto shape_c = Convert2SizeTClipNeg(std::get<IDX_OUT_SHAPE>(res_op_[idx_op])); in ResOpForward() local
379 auto shape_c = std::get<IDX_OUT_SHAPE>(res_op_[idx_op - 1]); in ResOpForward() local
Deinsum_gpu_kernel.h87 auto shape_c = Convert2SizeTClipNeg(std::get<IDX_OUT_SHAPE>(res_op_[count])); in Launch() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/fusion/
Dtf_lstm_cell_fusion.cc273 std::vector<int64_t> shape_c{1, kUnidirectionalGateNum * hidden_size, hidden_size}; in SplitWeights() local
/third_party/mindspore/mindspore-src/source/mindspore/core/ir/
Dbase_tensor.h803 ShapeVector shape_c(void) const { return shape(); } in shape_c() function