Home
last modified time | relevance | path

Searched defs:is_tensor (Results 1 – 8 of 8) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/delegate/tensorrt/op/
Dreduce_tensorrt.cc74 bool is_tensor = out_tensor->getDimensions().nbDims != 0; in AddInnerOp() local
Delementwise_tensorrt.cc175 auto is_tensor = x_input.is_tensor || y_input.is_tensor; in AddInnerOp() local
/third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/ops_generate/
Dpyboost_utils.py28 def is_tensor(op_arg): function
/third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/parallel/_transformer/
Dlayers.py183 def _check_past_none_input_none(use_past, param_name, func_name, default_value, is_tensor, is_defau… argument
/third_party/mindspore/mindspore-src/source/mindspore/core/abstract/ops/
Dprimitive_infer_map.cc74 auto is_tensor = input_abstract->isa<abstract::AbstractTensor>(); in CheckNeedAddToDependList() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/session/
Dsession_basic.cc855 auto is_tensor = input_value->isa<tensor::Tensor>(); in GetOpInputTensors() local
/third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/ops/function/
Darray_func.py4376 def is_tensor(obj): function
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/pipeline/jit/ps/
Dpipeline.cc723 bool is_tensor = false; in VerifyInputSignature() local