Home
last modified time | relevance | path

Searched defs:Tensor (Results 1 – 25 of 105) sorted by relevance

12345

/third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/hypercomplex/hypercomplex/
D_hc_bn_impl.py45 gamma_init: Union[Tensor, str, Initializer, numbers.Number],
46 beta_init: Union[Tensor, str, Initializer, numbers.Number],
51 def get_norm(self, u: Tensor) -> Tensor:
68 def get_square_norm(self, u: Tensor) -> Tensor:
86 u_x: Tensor,
87 u_y: Tensor,
88 scale_x: Tensor,
89 scale_y: Tensor,
90 shift_x: Tensor,
91 shift_y: Tensor) -> Tuple[Tensor, Tensor]:
[all …]
Dhc_pool.py123 def construct(self, u: Tensor) -> Tensor:
134 x: Tensor,
135 y: Tensor) -> Tuple[Tensor, Tensor]:
222 x: Tensor,
223 y: Tensor) -> Tuple[Tensor, Tensor]:
319 x: Tensor,
320 y: Tensor) -> Tuple[Tensor, Tensor]:
419 x: Tensor,
420 y: Tensor) -> Tuple[Tensor, Tensor]:
518 x: Tensor,
[all …]
Dhc_bn.py121 gamma_init: Union[Tensor, str, Initializer, numbers.Number] = 'ones',
122 beta_init: Union[Tensor, str, Initializer, numbers.Number] = 'zeros',
123 moving_mean_init: Union[Tensor, str, Initializer, numbers.Number] = 'zeros',
124 moving_var_init: Union[Tensor, str, Initializer, numbers.Number] = 'ones',
172 def construct(self, u: Tensor) -> Tensor:
231 u_centered: Tensor,
232 sigma2: Tensor,
252 u: Tensor,
359 gamma_init: Union[Tensor, str, Initializer, numbers.Number] = 'ones',
360 beta_init: Union[Tensor, str, Initializer, numbers.Number] = 'zeros',
[all …]
Dhc_conv.py164 weight_init: Union[Tensor, str, Initializer, numbers.Number],
165 bias_init: Union[Tensor, str, Initializer, numbers.Number],
218 weight_init: Union[Tensor, str, Initializer, numbers.Number],
219 bias_init: Union[Tensor, str, Initializer, numbers.Number],
268 def construct(self, u: Tensor) -> Tensor:
299 x: Tensor,
300 y: Tensor) -> Tuple[Tensor, Tensor]:
458 weight_init: Union[Tensor, str, Initializer, numbers.Number] = 'normal',
459 bias_init: Union[Tensor, str, Initializer, numbers.Number] = 'zeros',
481 x: Tensor,
[all …]
D_hc_dense_impl.py62 weight_init: Union[Tensor, str, Initializer, numbers.Number],
68 x: Tensor,
69 y: Tensor) -> Tuple[Tensor, Tensor]:
102 weight_init: Union[Tensor, str, Initializer, numbers.Number],
D_hc_conv_impl.py66 weight_init: Union[Tensor, str, Initializer, numbers.Number],
73 x: Tensor,
74 y: Tensor,
116 weight_init: Union[Tensor, str, Initializer, numbers.Number],
Dhc_dense.py95 weight_init: Union[Tensor, str, Initializer, numbers.Number] = 'normal',
96 bias_init: Union[Tensor, str, Initializer, numbers.Number] = 'zeros',
159 def check_dense_input_shape(self, x: Tensor, x_dtype):
168 def construct(self, u: Tensor) -> Tensor:
/third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/hypercomplex/double/
D_double_bn_impl.py52 u_x: Tensor,
53 u_y: Tensor,
54 scale_x: Tensor,
55 scale_y: Tensor,
56 shift_x: Tensor,
57 shift_y: Tensor) -> Tuple[Tensor, Tensor]:
89 def get_norm(self, u: Tensor) -> Tensor:
114 def get_square_norm(self, u: Tensor) -> Tensor:
165 u_x: Tensor,
166 u_y: Tensor,
[all …]
Ddouble_operators.py239 weight_init: Union[Tensor, str, Initializer, numbers.Number] = 'normal',
240 bias_init: Union[Tensor, str, Initializer, numbers.Number] = 'zeros',
435 weight_init: Union[Tensor, str, Initializer, numbers.Number] = 'normal',
436 bias_init: Union[Tensor, str, Initializer, numbers.Number] = 'zeros',
668 weight_init: Union[Tensor, str, Initializer, numbers.Number] = 'normal',
669 bias_init: Union[Tensor, str, Initializer, numbers.Number] = 'zeros',
817 gamma_init: Union[Tensor, str, Initializer, numbers.Number] = 'ones',
818 beta_init: Union[Tensor, str, Initializer, numbers.Number] = 'zeros',
819 moving_mean_init: Union[Tensor, str, Initializer, numbers.Number] = 'zeros',
820 moving_var_init: Union[Tensor, str, Initializer, numbers.Number] = 'ones',
[all …]
D_double_dense_impl.py63 real: Tensor,
64 double: Tensor) -> Tuple[Tensor, Tensor]:
116 u1: Tensor,
117 u2: Tensor) -> Tuple[Tensor, Tensor]:
D_double_conv_impl.py65 real: Tensor,
66 double: Tensor,
127 u1: Tensor,
128 u2: Tensor,
/third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/nn/layer/
Dtransformer.py194 …def construct(self, query: Tensor, key: Tensor, value: Tensor, key_padding_mask: Optional[Tensor] …
195 … need_weights: bool = True, attn_mask: Optional[Tensor] = None, average_attn_weights: bool = True):
355 def construct(self, src: Tensor, src_mask: Optional[Tensor] = None,
356 src_key_padding_mask: Optional[Tensor] = None):
507 def construct(self, tgt: Tensor, memory: Tensor, tgt_mask: Optional[Tensor] = None,
508 … memory_mask: Optional[Tensor] = None, tgt_key_padding_mask: Optional[Tensor] = None,
509 memory_key_padding_mask: Optional[Tensor] = None):
595 …def construct(self, src: Tensor, src_mask: Optional[Tensor] = None, src_key_padding_mask: Optional…
670 def construct(self, tgt: Tensor, memory: Tensor, tgt_mask: Optional[Tensor] = None,
671 … memory_mask: Optional[Tensor] = None, tgt_key_padding_mask: Optional[Tensor] = None,
[all …]
/third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/hypercomplex/complex/
D_complex_bn_impl.py49 u_x: Tensor,
50 u_y: Tensor,
51 scale_x: Tensor,
52 scale_y: Tensor,
53 shift_x: Tensor,
54 shift_y: Tensor) -> Tuple[Tensor, Tensor]:
87 u: Tensor) -> Tensor:
116 u: Tensor) -> Tensor:
Dcomplex_operators.py232 weight_init: Union[Tensor, str, Initializer, numbers.Number] = 'normal',
233 bias_init: Union[Tensor, str, Initializer, numbers.Number] = 'zeros',
424 weight_init: Union[Tensor, str, Initializer, numbers.Number] = 'normal',
425 bias_init: Union[Tensor, str, Initializer, numbers.Number] = 'zeros',
651 weight_init: Union[Tensor, str, Initializer, numbers.Number] = 'normal',
652 bias_init: Union[Tensor, str, Initializer, numbers.Number] = 'zeros',
775 gamma_init: Union[Tensor, str, Initializer, numbers.Number] = 'ones',
776 beta_init: Union[Tensor, str, Initializer, numbers.Number] = 'zeros',
777 moving_mean_init: Union[Tensor, str, Initializer, numbers.Number] = 'zeros',
778 moving_var_init: Union[Tensor, str, Initializer, numbers.Number] = 'ones',
[all …]
D_complex_dense_impl.py60 real: Tensor,
61 imag: Tensor) -> Tuple[Tensor, Tensor]:
115 real: Tensor,
116 imag: Tensor) -> Tuple[Tensor, Tensor]:
D_complex_conv_impl.py67 real: Tensor,
68 imag: Tensor,
133 real: Tensor,
134 imag: Tensor,
198 weight_init: Union[Tensor, str, Initializer, numbers.Number],
209 real: Tensor,
210 imag: Tensor,
/third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/hypercomplex/dual/
D_dual_bn_impl.py49 u_x: Tensor,
50 u_y: Tensor,
51 scale_x: Tensor,
52 scale_y: Tensor,
53 shift_x: Tensor,
54 shift_y: Tensor) -> Tuple[Tensor, Tensor]:
86 def get_norm(self, u: Tensor) -> Tensor:
117 def get_square_norm(self, u: Tensor) -> Tensor:
Ddual_operators.py206 weight_init: Union[Tensor, str, Initializer, numbers.Number] = 'normal',
207 bias_init: Union[Tensor, str, Initializer, numbers.Number] = 'zeros',
358 weight_init: Union[Tensor, str, Initializer, numbers.Number] = 'normal',
359 bias_init: Union[Tensor, str, Initializer, numbers.Number] = 'zeros') -> None:
549 weight_init: Union[Tensor, str, Initializer, numbers.Number] = 'normal',
550 bias_init: Union[Tensor, str, Initializer, numbers.Number] = 'zeros',
652 gamma_init: Union[Tensor, str, Initializer, numbers.Number] = 'ones',
653 beta_init: Union[Tensor, str, Initializer, numbers.Number] = 'zeros',
654 moving_mean_init: Union[Tensor, str, Initializer, numbers.Number] = 'zeros',
655 moving_var_init: Union[Tensor, str, Initializer, numbers.Number] = 'ones',
[all …]
D_dual_conv_impl.py66 real: Tensor,
67 dual: Tensor,
127 weight_init: Union[Tensor, str, Initializer, numbers.Number],
138 real: Tensor,
139 dual: Tensor,
D_dual_dense_impl.py61 real: Tensor,
62 dual: Tensor) -> Tuple[Tensor, Tensor]:
/third_party/mindspore/mindspore-src/source/mindspore/core/ir/
Dtensor.cc138 Tensor::Tensor(const Tensor &tensor) in Tensor() function in mindspore::tensor::Tensor
150 Tensor::Tensor(const Tensor &tensor, TypeId data_type) in Tensor() function in mindspore::tensor::Tensor
162 Tensor::Tensor(const BaseTensor &tensor, TypeId data_type) : BaseTensor(tensor, data_type) {} in Tensor() function in mindspore::tensor::Tensor
164 Tensor::Tensor(const BaseTensor &base_tensor) : BaseTensor(base_tensor) {} in Tensor() function in mindspore::tensor::Tensor
188 Tensor::Tensor(TypeId data_type, const ShapeVector &shape, TensorDataPtr data) : BaseTensor(data_ty… in Tensor() function in mindspore::tensor::Tensor
190 Tensor::Tensor(TypeId data_type, const ShapeVector &shape) : BaseTensor(data_type, shape) {} in Tensor() function in mindspore::tensor::Tensor
192 Tensor::Tensor(TypeId data_type, const ShapeVector &shape, void *data, size_t data_len) in Tensor() function in mindspore::tensor::Tensor
195 Tensor::Tensor(TypeId data_type, const ShapeVector &shape, void *data, TypeId src_data_type) in Tensor() function in mindspore::tensor::Tensor
198 Tensor::Tensor(const std::vector<int64_t> &input, const TypePtr &data_type) : BaseTensor(input, dat… in Tensor() function in mindspore::tensor::Tensor
200 Tensor::Tensor(const std::vector<int32_t> &input, const TypePtr &data_type) : BaseTensor(input, dat… in Tensor() function in mindspore::tensor::Tensor
[all …]
/third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/ops/function/
Dsparse_func.py107 def coalesce(x_indices: Tensor, x_values: Tensor, x_shape: Tensor) -> Tuple[Tensor, Tensor, Tensor]:
169 def csr_div(x: CSRTensor, y: Tensor) -> Tensor:
203 def csr_mul(x: CSRTensor, y: Tensor) -> CSRTensor:
228 def csr_mv(csr_tensor: CSRTensor, dense: Tensor) -> Tensor:
446 def dense_to_sparse_coo(tensor: Tensor) -> COOTensor:
492 def dense_to_sparse_csr(tensor: Tensor) -> CSRTensor:
656 def coo_add(x1: COOTensor, x2: COOTensor, thresh: Tensor) -> COOTensor:
774 def csr_add(a: CSRTensor, b: CSRTensor, alpha: Tensor, beta: Tensor) -> CSRTensor:
/third_party/rust/rust/tests/ui/const-generics/issues/
Dissue-83288.rs43 pub struct Tensor<I: Indices<N>, const N: usize> struct
51 impl<I: Indices<N>, J: Indices<N>, const N: usize> Mul<Tensor<J, N>> for Tensor<I, N> implementation
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/
Dmindrt_executor.h34 explicit MindrtExecutor(std::unordered_map<Tensor *, Tensor *> *output_map, in MindrtExecutor()
/third_party/mindspore/mindspore-src/source/tests/st/rewrite/
Dmodels.py27 def construct(self, x: Optional[Tensor]):
40 def no_cell_func(self, x: Optional[Tensor]):
63 def class_var_func(self, x: Optional[Tensor]):

12345