Home
last modified time | relevance | path

Searched defs:GetShape (Results 1 – 25 of 26) sorted by relevance

12

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/graph_kernel/expander/mindir_adapter/
Danf_node_holder.cc35 ShapeVector AnfNodeHolderWithDeviceInfo::GetShape() { return Callback::Instance()->GetOutputShape(n… in GetShape() function in mindspore::graphkernel::expander::AnfNodeHolderWithDeviceInfo
39 ShapeVector AnfNodeHolderWithHostInfo::GetShape() { return Callback::Instance()->GetOutputInferShap… in GetShape() function in mindspore::graphkernel::expander::AnfNodeHolderWithHostInfo
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/common/expander/core/
Dnode.cc31 BaseShapePtr Node::GetShape() { MS_EXCEPTION(NotImplementedError) << "Base node not implement GetSh… in GetShape() function in mindspore::expander::Node
127 BaseShapePtr IrNode::GetShape() { in GetShape() function in mindspore::expander::IrNode
164 BaseShapePtr FuncNode::GetShape() { in GetShape() function in mindspore::expander::FuncNode
Dinfer.cc73 BaseShapePtr CppInfer::GetShape(const NodePtr &node) { in GetShape() function in mindspore::expander::CppInfer
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/minddata/dataset/core/
Ddevice_tensor_ascend910b.h63 TensorShape &GetShape() { return tensor_shape_; } in GetShape() function
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/transform/graph_ir/
Dstorage_format_operator.cc29 inline ShapeVector GetShape(const AnfNodePtr &node) { in GetShape() function
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/
Dsequence_addn.cc27 std::vector<int64_t> GetShape(const ::aicpuops::TensorShape &shape) { in GetShape() function
Dsequence_stack.cc32 std::vector<int64_t> GetShape(const ::aicpuops::TensorShape &shape) { in GetShape() function
Dsequence_concat.cc27 std::vector<int64_t> GetShape(const ::aicpuops::TensorShape &shape) { in GetShape() function
Dslice_grad_kernel.cc33 std::vector<int64_t> GetShape(const ::aicpuops::TensorShape &shape) { in GetShape() function
Dquant_dtype_cast_kernel.cc36 std::vector<int64_t> GetShape(const ::aicpuops::TensorShape &shape) { in GetShape() function
Dfse_decode_kernel.cc117 std::vector<int> GetShape(const ::aicpuops::TensorShape &shape) { in GetShape() function
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/expander/fallback/
Dfallback_irbuilder.h59 ShapeVector GetShape(const NodePtr &node) const { return node->shape(); } in GetShape() function
/third_party/mindspore/mindspore-src/source/mindspore/core/symbolic_shape/
Doperation_builder.cc45 SymbolPtr OperationBuilder::GetShape(const AbstractBasePtr &abs) const { in GetShape() function in mindspore::symshape::OperationBuilder
/third_party/mindspore/mindspore-src/source/mindspore/core/ops/
Dedit_distance.cc89 auto GetShape = [&input_args](size_t index) { in EditDistanceInferShape() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/expander/bprop/
Dbprop_irbuilder.h63 ShapeVector GetShape(const NodePtr &node) const { return node->shape(); } in GetShape() function
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/
Dminimum_grad_cpu_kernel.cc47 void GetShape(std::vector<size_t> *shape, const ShapeVector &shape_, const ShapeVector &dout_shape)… in GetShape() function
Dmaximum_grad_cpu_kernel.cc171 void GetShape(std::vector<size_t> *shape, const ShapeVector &shape_, const ShapeVector &dout_shape)… in GetShape() function
Dunique_consecutive_cpu_kernel.cc72 std::vector<T> GetShape() { return shape_; } in GetShape() function in mindspore::kernel::__anonbd2bb2730111::PositionIterator
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/utils/
Dstub_tensor.cc150 py::object TensorNode::GetShape() { in GetShape() function in mindspore::stub::TensorNode
Danfalgo.cc722 inline ShapeVector GetShape(const abstract::BaseShapePtr &base_shape) { in GetShape() function
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/js_api/
Dmstensor_napi.cc181 napi_value MSTensorNapi::GetShape(napi_env env, napi_callback_info info) { in GetShape() function in mindspore::MSTensorNapi
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/include/backend/debug/
Dtensor_data.h213 std::vector<int64_t> GetShape() const { return this->shape_; } in GetShape() function
/third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/ops/operations/
D_ocr_ops.py23 class GetShape(PrimitiveWithInfer): class
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/graph_kernel/
Dgraph_kernel_helper.cc209 ShapeVector GetShape(const AnfNodePtr &node) { in GetShape() function
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/cpu_kernel/ms_kernel/
Dunique_consecutive.cc79 std::vector<T> GetShape() { return shape_; } in GetShape() function in __anoncb0ead6b0111::PositionIterator

12