Searched refs:out_ms (Results 1 – 9 of 9) sorted by relevance
/third_party/mindspore/tests/st/pynative/ |
D | test_pynative_hook_grad.py | 235 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 …]
|
D | test_tensor_setitem.py | 39 out_ms = TensorSetItem()(input_ms) 41 assert onp.all(out_ms.asnumpy() == out_np)
|
/third_party/mindspore/tests/st/ops/cpu/ |
D | test_sparse_tensor_dense_matmul_op.py | 66 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/ |
D | model_impl.cc | 42 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/ |
D | test_maximum.py | 53 out_ms = me_max(a, b) 56 print("ms out :{}".format(out_ms))
|
/third_party/mindspore/tests/st/control/ |
D | test_if_by_if.py | 66 out_ms = net(Tensor(input_np)) 68 assert np.allclose(out_ms.asnumpy(), out_np)
|
/third_party/mindspore/tests/ut/python/dataset/ |
D | test_time_stretch.py | 122 out_ms = c_audio.TimeStretch(64, 2, 1.6)(spectrogram) 124 allclose_nparray(out_ms, out_expect, 0.001, 0.001)
|
D | test_frequency_masking.py | 73 out_ms = audio.FrequencyMasking(False, 2, 0, 0)(spectrogram) 80 allclose_nparray(out_ms, out_benchmark, 0.0001, 0.0001)
|
D | test_time_masking.py | 73 out_ms = audio.TimeMasking(False, 2, 0, 0)(spectrogram) 80 allclose_nparray(out_ms, out_benchmark, 0.0001, 0.0001)
|