Home
last modified time | relevance | path

Searched refs:out_ms (Results 1 – 9 of 9) sorted by relevance

/third_party/mindspore/tests/st/pynative/
Dtest_pynative_hook_grad.py235 out_ms = ms_net(input_ms, Tensor(1))
238 grad_net(input_ms, Tensor(1), out_ms)
247 out_ms = ms_net(input_ms)
250 input_ms_grad = grad_net(input_ms, out_ms)
270 out_ms = ms_net(input_ms)
273 grad_net(input_ms, out_ms)
296 out_ms = ms_net(input_ms)
299 input_ms_grad = grad_net(input_ms, out_ms)
314 out_ms = ms_net(input1_ms, input2_ms)
317 input_ms_grad = grad_net(input1_ms, input2_ms, out_ms)
[all …]
Dtest_tensor_setitem.py39 out_ms = TensorSetItem()(input_ms)
41 assert onp.all(out_ms.asnumpy() == out_np)
/third_party/mindspore/tests/st/ops/cpu/
Dtest_sparse_tensor_dense_matmul_op.py66 out_ms = sparse_dense_matmul_net(indices, values, dense_shape, dense)
68 judge_result_correct(out_ms.asnumpy(), out_np)
95 out_ms = sparse_dense_matmul_net(indices, values, dense_shape, dense)
98 judge_result_correct(out_ms.asnumpy(), out_np)
125 out_ms = sparse_dense_matmul_net(indices, values, dense_shape, dense)
128 judge_result_correct(out_ms.asnumpy(), out_np)
155 out_ms = sparse_dense_matmul_net(indices, values, dense_shape, dense)
158 judge_result_correct(out_ms.asnumpy(), out_np)
/third_party/mindspore/mindspore/lite/src/cxx_api/train/
Dmodel_impl.cc42 Status ModelImpl::PrepareMetrics(Model *model, std::vector<session::Metrics *> *out_ms, in PrepareMetrics() argument
44 if (out_ms == nullptr || adapter_ms == nullptr) { in PrepareMetrics()
62 out_ms->push_back(internal_m); in PrepareMetrics()
71 out_ms->push_back(adapter_m); in PrepareMetrics()
/third_party/mindspore/tests/st/ops/ascend/test_tbe_ops/
Dtest_maximum.py53 out_ms = me_max(a, b)
56 print("ms out :{}".format(out_ms))
/third_party/mindspore/tests/st/control/
Dtest_if_by_if.py66 out_ms = net(Tensor(input_np))
68 assert np.allclose(out_ms.asnumpy(), out_np)
/third_party/mindspore/tests/ut/python/dataset/
Dtest_time_stretch.py122 out_ms = c_audio.TimeStretch(64, 2, 1.6)(spectrogram)
124 allclose_nparray(out_ms, out_expect, 0.001, 0.001)
Dtest_frequency_masking.py73 out_ms = audio.FrequencyMasking(False, 2, 0, 0)(spectrogram)
80 allclose_nparray(out_ms, out_benchmark, 0.0001, 0.0001)
Dtest_time_masking.py73 out_ms = audio.TimeMasking(False, 2, 0, 0)(spectrogram)
80 allclose_nparray(out_ms, out_benchmark, 0.0001, 0.0001)