| /third_party/mindspore/mindspore/explainer/ |
| D | _operators.py | 54 Tensor = mindspore.Tensor variable
57 def absolute(inputs: Tensor) -> Tensor:
75 def argmax(inputs: Tensor, axis: int = -1, keep_dims: bool = False) -> Tensor:
86 def argmin(inputs: Tensor, axis: int = -1, keep_dims: bool = False) -> Tensor:
97 def argsort(inputs: Tensor, axis: int = -1, descending: bool = False) -> Tensor:
106 def assign(inputs: Tensor, idx: _Idx, value: Tensor) -> Tensor:
117 def intersection(*inputs: Tensor) -> Tensor:
126 def matmul(inputs_x: Tensor, inputs_y: Tensor) -> Tensor:
133 def maximum(inputs: Tensor, axis: _Axis = (), keep_dims: bool = False) -> Tensor:
140 def minimum(inputs: Tensor, axis: _Axis = (), keep_dims: bool = False) -> Tensor:
[all …]
|
| D | _utils.py | 217 def format_tensor_to_ndarray(x: Union[ms.Tensor, np.ndarray]) -> np.ndarray:
228 def calc_correlation(x: Union[ms.Tensor, np.ndarray],
229 y: Union[ms.Tensor, np.ndarray]) -> float:
|
| /third_party/mindspore/mindspore/core/ir/ |
| D | tensor.cc | 487 Tensor::Tensor(const Tensor &tensor) in Tensor() function in mindspore::tensor::Tensor
503 Tensor::Tensor(const Tensor &tensor, TypeId data_type) in Tensor() function in mindspore::tensor::Tensor
519 Tensor::Tensor(TypeId data_type, const ShapeVector &shape, TensorDataPtr data) in Tensor() function in mindspore::tensor::Tensor
522 Tensor::Tensor(TypeId data_type, const ShapeVector &shape) in Tensor() function in mindspore::tensor::Tensor
525 Tensor::Tensor(TypeId data_type, const ShapeVector &shape, void *data, size_t data_len) in Tensor() function in mindspore::tensor::Tensor
528 Tensor::Tensor(TypeId data_type, const ShapeVector &shape, void *data, TypeId src_data_type) in Tensor() function in mindspore::tensor::Tensor
531 Tensor::Tensor(const std::vector<int64_t> &input, const TypePtr &data_type) in Tensor() function in mindspore::tensor::Tensor
536 Tensor::Tensor(const std::vector<double> &input, const TypePtr &data_type) in Tensor() function in mindspore::tensor::Tensor
541 Tensor::Tensor(int64_t input, const TypePtr &data_type) in Tensor() function in mindspore::tensor::Tensor
546 Tensor::Tensor(double input, const TypePtr &data_type) in Tensor() function in mindspore::tensor::Tensor
[all …]
|
| /third_party/mindspore/tests/st/auto_monad/ |
| D | test_auto_monad_layer.py | 31 def reconstruct_sample(self, x: Tensor):
37 def construct(self, x: Tensor):
|
| /third_party/mindspore/mindspore/lite/src/ |
| D | sub_graph_split.h | 44 struct Tensor { struct
47 TensorType type_; argument
|
| D | mindrt_executor.h | 32 …explicit MindrtExecutor(std::unordered_map<Tensor *, Tensor *> *output_map) : output_tensor_map_(o… in MindrtExecutor()
|
| D | tensor.cc | 30 Tensor::Tensor(const TypeId data_type, std::vector<int> shape, const mindspore::Format &format, Cat… in Tensor() function in mindspore::lite::Tensor
|
| D | scheduler.cc | 756 int CastAndRestoreConstTensorData(Tensor *tensor, std::map<Tensor *, Tensor *> *restored_origin_ten… in CastAndRestoreConstTensorData()
793 int CastConstTensorsData(const std::vector<Tensor *> &tensors, std::map<Tensor *, Tensor *> *restor… in CastConstTensorsData()
830 inline void FreeRestoreTensors(std::map<Tensor *, Tensor *> *restored_origin_tensors) { in FreeRestoreTensors()
839 inline void RestoreTensorData(std::map<Tensor *, Tensor *> *restored_origin_tensors) { in RestoreTensorData()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/aicpu/proto/ |
| D | tensor.proto | 23 message Tensor { message
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/core/ |
| D | device_tensor.h | 30 class Tensor; variable
|
| D | global_context.h | 31 class Tensor; variable
|
| /third_party/mindspore/tests/ut/python/train/summary/ |
| D | test_histogram_summary.py | 35 def _wrap_test_data(input_data: Tensor):
|
| /third_party/mindspore/mindspore/ccsrc/utils/ |
| D | convert_utils.h | 37 class Tensor; variable
|
| /third_party/mindspore/mindspore/lite/test/ut/src/runtime/kernel/opencl/ |
| D | common.h | 32 using Tensor = mindspore::lite::Tensor; variable
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/include/dataset/ |
| D | iterator.h | 35 class Tensor; variable
|
| D | datasets.h | 44 class Tensor; variable
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/tbe_kernel_select/ |
| D | tbe_property_checker.cc | 25 using Tensor = mindspore::tensor::Tensor; typedef
|
| /third_party/mindspore/tests/ut/cpp/pynative/ |
| D | pynative_execute_test.cc | 28 using Tensor = mindspore::tensor::Tensor; typedef
|
| /third_party/mindspore/mindspore/lite/micro/coder/allocator/ |
| D | allocator.cc | 84 void MemoryAllocator::RecordTensorsAddr(const std::map<Tensor *, size_t> &offsets) { in RecordTensorsAddr()
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/opencl/kernel/ |
| D | fusion_eltwise.cc | 81 … const std::map<lite::Tensor *, FusionEltwiseParameter *> &replace_map = {}) { in CreateParam()
130 LiteKernel *node, const std::map<lite::Tensor *, FusionEltwiseParameter *> &replace_map) { in CreateFusionEltwiseParameter()
|
| /third_party/mindspore/tests/ut/cpp/vm/ |
| D | segment_runner_test.cc | 38 using Tensor = tensor::Tensor; typedef
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/liteapi/include/ |
| D | datasets.h | 42 class Tensor; variable
|
| /third_party/mindspore/mindspore/lite/micro/coder/ |
| D | context.h | 45 void set_tensor_map(const std::map<Tensor *, std::string> &tensor_map) { in set_tensor_map()
|
| /third_party/mindspore/mindspore/profiler/parser/ |
| D | memory_usage_parser.py | 29 from mindspore.profiler.parser.container import MemoryTensor as Tensor unknown
|
| /third_party/mindspore/mindspore/ccsrc/transform/graph_ir/ |
| D | op_adapter_util.cc | 30 GeTensor ConvertAnyUtil(const ValuePtr &value, const AnyTraits<mindspore::tensor::Tensor> &) { in ConvertAnyUtil()
|