Home
last modified time | relevance | path

Searched defs:inputTensor (Results 1 – 9 of 9) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/auto_parallel/rec_core/
Drec_strategy.h36 TensorStr4D inputTensor[MAX_INPUT_NUM]; member
/third_party/mindspore/mindspore-src/source/mindspore/lite/examples/cloud_infer/quick_start_java/src/main/java/com/mindspore/lite/demo/
DMain.java79 MSTensor inputTensor = model.getInputs().get(0); in run() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/examples/quick_start_java/src/main/java/com/mindspore/lite/demo/
DMain.java78 MSTensor inputTensor = model.getInputByTensorName("graph_input-173"); in run() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/examples/runtime_java/app/src/main/java/com/mindspore/lite/demo/
DMainActivity.java160 MSTensor inputTensor = model.getInputByTensorName("graph_input-173"); in runInference() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/cpu_kernel/ms_kernel/
Duniform.cc37 Tensor *inputTensor = ctx.Input(0); in Compute() local
Dscale_and_translate.cc395 EigenTensor inputTensor(input, input->GetData()); in ScaleAndTranslateCompute() local
458 EigenTensor inputTensor(input, input->GetData()); in ScaleAndTranslateGradCompute() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/
Dscale_and_translate_cpu_kernel.cc399 EigenTensor inputTensor(input0_shape_, input); in LaunchKernel() local
448 EigenTensor inputTensor(input0_shape_, input); in LaunchKernel() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/expander/fallback/
Dnn_ops.cc723 bool IsInputNeedExpand(NodePtr paddingsTensor, NodePtr inputTensor) { in IsInputNeedExpand()
/third_party/mindspore/mindspore-src/source/third_party/proto/coreml/
DNeuralNetwork.proto464 repeated Tensor inputTensor = 4; // must be the same length as the "input" field field