Home
last modified time | relevance | path

Searched defs:shape_vector (Results 1 – 25 of 78) sorted by relevance

1234

/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/parser/onnx/
Donnx_given_tensor_fill_parser.cc43 ShapeVector shape_vector(shape.begin(), shape.end()); in ParseInt8GivenIntTensorFill() local
71 ShapeVector shape_vector(shape.begin(), shape.end()); in ParseInt8GivenTensorFill() local
85 std::vector<int64_t> shape_vector; in Parse() local
Donnx_constant_parser.cc54 …std::vector<int64_t> shape_vector(onnx_const_tensor.dims().begin(), onnx_const_tensor.dims().end()… in AddDataInfoAttr() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/adapter/dpico/src/
Ddpico_preprocess_pass.cc32 … const ShapeVector &shape_vector, const api::AbstractBasePtr &abstract) { in InsertTransposeBeforeBiasAdd()
98 const ShapeVector &shape_vector) { in InsertTransposeAfterBiasAdd()
142 ShapeVector shape_vector; in PreProcessBiadAdd() local
Dcustom_creator.cc155 ShapeVector shape_vector = {static_cast<int64_t>(om_model_info->modelSize)}; in CreateOmParameter() local
210 ShapeVector shape_vector(input_info.shape.begin(), input_info.shape.end()); in SetSubgraphInputOutputDims() local
222 ShapeVector shape_vector(output_info.shape.begin(), output_info.shape.end()); in SetSubgraphInputOutputDims() local
340 ShapeVector shape_vector(output_info.shape.begin(), output_info.shape.end()); in SetCustomSingleOutput() local
384 ShapeVector shape_vector(output_info.shape.begin(), output_info.shape.end()); in SetCustomMultiOutput() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/adapter/dpico/common/
Danf_util.cc108 STATUS GetShapeVectorFromParameter(const api::AnfNodePtr &anode, ShapeVector *shape_vector) { in GetShapeVectorFromParameter()
204 ShapeVector shape_vector; in GetOutputShapesFromCNode() local
217 ShapeVector shape_vector; in GetOutputShapesFromCNode() local
296 int GetAnfNodeOutputShape(const api::AnfNodePtr &input, ShapeVector *shape_vector) { in GetAnfNodeOutputShape()
546 std::vector<int64_t> shape_vector{static_cast<int64_t>(data.size())}; in BuildIntVecParameterNode() local
570 std::vector<int64_t> shape_vector; in BuildIntVec2DParameterNode() local
702 …taTypeAndShape(const api::ParameterPtr &param_node, TypeId *data_type, ShapeVector *shape_vector) { in GetDataTypeAndShape()
737 STATUS GetShapeVectorFromStringTensor(const api::TensorPtr &tensor_info, ShapeVector *shape_vector,… in GetShapeVectorFromStringTensor()
Dfetch_content.cc32 ShapeVector shape_vector; in FetchFromDefaultParam() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/graph_scheduler/actor/rpc/protocol/
Drpc.proto21 repeated int64 shape_vector = 1; field
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/lite_exporter/
Dfetch_content.cc49 …apeVectorFromStringTensor(const tensor::TensorPtr &tensor_info, ShapeVector *shape_vector, size_t … in GetShapeVectorFromStringTensor()
104 …GetDataTypeAndShape(const ParameterPtr &param_node, TypeId *data_type, ShapeVector *shape_vector) { in GetDataTypeAndShape()
141 auto shape_vector = utils::cast<abstract::ShapePtr>(abstract_tensor->BuildShape())->shape(); in FetchFromTensorValue() local
287 ShapeVector shape_vector; in FetchFromDefaultParam() local
421 auto shape_vector = utils::cast<abstract::ShapePtr>(abstract_tensor->BuildShape())->shape(); in FetchDataFromCNode() local
491 auto shape_vector = utils::cast<abstract::ShapePtr>(abstract_tensor->BuildShape())->shape(); in FetchDataFromAbstract() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/graph_compiler/
Danfnode_tensor_adapter.cc172 ShapeVector shape_vector; in Create() local
448 ShapeVector shape_vector; in Create() local
463 ShapeVector *shape_vector) { in GetDTAndShapeFromAbTensor()
518 ShapeVector *shape_vector) { in GetDTAndShapeFromParameter()
538 ShapeVector shape_vector; in SetDTAndShapeFromAbTensorToLiteTensor() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/adapter/acl/mapper/
Dmatmul_fusion_mapper.cc96 std::vector<int64_t> shape_vector; in Mapper() local
159 std::vector<int64_t> shape_vector; in QuantMapper() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/test/common/
Dimport_from_meta_graphT.cc53 std::vector<int64_t> shape_vector(shape.begin(), shape.end()); in ConverterConstTensor() local
98 std::vector<int64_t> shape_vector; in ConvertTensorToAbstractTensor() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/pyboost/customize/
Drand_ext.cc32 std::vector<int64_t> shape_vector = ConvertValueTupleToVector<int64_t>(shape); in RandExtAscendCustomize() local
Ddropout_gen_mask_ext.cc31 auto shape_vector = ConvertValueTupleToVector<int64_t>(shape); in DropoutGenMaskExtAscendCustomize() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/kernel/default/
Dcnode_infer_manager.cc29 ShapeVector shape_vector; in SetDTAndShapeFromAbTensorToLiteTensor() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/customize/op_proto/
Delewise_calculation_ops_proto.cc186 std::vector<int64_t> shape_vector = op.GetDynamicInputDesc("x", 0).GetShape().GetDims(); in IMPLEMT_COMMON_INFERFUNC() local
194 std::vector<int64_t> shape_vector = {-2}; in IMPLEMT_COMMON_INFERFUNC() local
203 std::vector<int64_t> shape_vector = op.GetDynamicInputDesc("x", 0).GetShape().GetDims(); in IMPLEMT_COMMON_INFERFUNC() local
Dbroadcast_to_proto.cc50 std::vector<int64_t> shape_vector; in IMPLEMT_INFERFUNC() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/adapter/dpico/mapper/
Dbias_mapper.cc50 ShapeVector shape_vector; in SetBiasDataInfo() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/hal/device/
Dcpu_hash_table_util.h46 auto shape_vector = user_data->get<ShapeVector>(kHashTableShapeVector); in CreateCPUHashTable() local
Dcpu_device_address.h52 …CPUDeviceAddress(void *ptr, size_t size, const ShapeVector &shape_vector, const Format &format, Ty… in CPUDeviceAddress()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/
Dsize_cpu_kernel.cc44 auto shape_vector = inputs[kIndex0]->GetShapeVector(); in Resize() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/hal/device/
Dgpu_hash_table_util.h44 auto shape_vector = user_data->get<ShapeVector>(kHashTableShapeVector); in SetHashTable() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/kernel/
Dkernel.cc170 ShapeVector shape_vector = {SizeToLong(sequence_shape->size())}; in GetShapeVectorByBaseShape() local
274 void KernelTensor::SetShapeVector(const ShapeVector &shape_vector) { in SetShapeVector()
297 void KernelTensor::SetShapeVector(ShapeVector &&shape_vector) { in SetShapeVector()
/third_party/mindspore/mindspore-src/source/mindspore/core/abstract/
Dparam_validator.cc136 const auto &shape_vector = shape->GetShapeVector(); in CheckShapeSame() local
180 auto shape_vector = shape->shape(); in CheckShapeSame() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/graph/
Dgroup_depthwise_op_convert_pass.cc117 std::vector<int64_t> shape_vector; in Run() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/ge/
Davg_pool_grad_for_ge.cc50 auto shape_vector = input_x_shape->cast<abstract::ShapePtr>()->shape(); in Process() local

1234