Home
last modified time | relevance | path

Searched defs:tensor_input (Results 1 – 11 of 11) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/pass/
Dconvert_const_scalar_to_tensor.cc47 auto tensor_input = std::make_shared<ValueNode>(tensor_ptr); in ConvertTensorInput() local
Dconvert_const_input_to_tensor_input.cc57 auto tensor_input = CreateTensorInput(kernel_graph, input_node); in ConstInputToTensorInput() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/ge/
Ddropout_gen_mask_depend.cc62 auto tensor_input = CreateTensorInput(kernel_graph, input_node); in Run() local
Dge_convert_const_input_to_tensor_input.cc78 auto tensor_input = CreateTensorInput(kernel_graph, input_node); in ConstInputToTensorInput() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/quantizer/
Dtensor_compressor.cc41 int TensorCompressor::DoBitPack(const size_t &bit_num, schema::TensorT *tensor_input) { in DoBitPack()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/customize/op_proto/
Dmatrix_calculation_ops_proto.cc116 Tensor tensor_input; in IMPLEMT_COMMON_INFERFUNC() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/lite_exporter/
Danf_exporter.cc157 auto tensor_input = GetTensorFromAllTensor(meta_graph, activate_index); in ConvertQuantParam() local
221 auto tensor_input = meta_graph->allTensors[activate_index].get(); in ConvertQuantParam() local
260 … const AnfNodePtr &input_node, mindspore::schema::TensorT *tensor_input) { in SetInputQuantParamToTensorT()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/graph_kernel/
Ddecrease_transfer_precision.cc268 auto tensor_input = node->cast<CNodePtr>()->input(index); in ProcessSon() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/optimizer/
Dneighbor_exchange_v2_fusion.cc142 auto tensor_input = std::make_shared<ValueNode>(tensor_ptr); in ConvertValueToTensor() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/graph_compiler/
Dbackend_base.cc118 auto tensor_input = flatten_value[index]; in PushTupleTensor() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/optimizer/
Dhelper.cc308 auto tensor_input = std::make_shared<ValueNode>(tensor_ptr); in CreateTensorInput() local