Home
last modified time | relevance | path

Searched defs:input_shape_vector (Results 1 – 6 of 6) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/core/ops/ops_func_impl/
Ddynamic_quant_ext.cc27 auto input_shape_vector = input_shape->GetShapeVector(); in InferShape() local
Dbinary_cross_entropy_with_logits.cc52 auto input_shape_vector = input_shape_ptr->GetShapeVector(); in InferShape() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/config_parser/
Dacl_option_param_parser.cc130 STATUS AclOptionParamParser::ParseInputShapeVector(const std::string &input_shape_vector, in ParseInputShapeVector()
Dconfig_file_parser.h88 std::string input_shape_vector; member
149 std::string input_shape_vector; member
/third_party/mindspore/mindspore-src/source/mindspore/core/ops/
Dreshape_ext.cc131 auto input_shape_vector = input_shape->GetShapeVector(); in ReshapeExtInferShape() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/
Dfse_decode_kernel.cc180 std::vector<int> input_shape_vector(intput_shape, intput_shape + input_shape_size_); in FixedBitHalfDequantTask() local