Home
last modified time | relevance | path

Searched defs:tensor_w (Results 1 – 4 of 4) sorted by relevance

/third_party/mindspore/mindspore-src/source/tests/st/pi_jit/construct_input/
Dtest_outermost_input_pijit.py42 tensor_w = Tensor(np.ones((2, 2), np.float32) * 4) variable
/third_party/mindspore/mindspore-src/source/tests/ut/python/pipeline/parse/
Dtest_outermost_net_pass_non_tensor_inputs.py65 tensor_w = Tensor(np.ones((2, 2), np.float32) * 4) variable
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/auto_parallel/rec_core/
Drec_cost.cc886 …int64_t tensor_w = static_cast<int64_t>(op.arguments[0].tensor_shape.shape_w * op.arguments[0].ten… in GetOptimalStr() local
969 …int64_t tensor_w = static_cast<int64_t>(op.arguments[0].tensor_shape.shape_w * op.arguments[0].ten… in GetOptimalStr() local
/third_party/mindspore/mindspore-src/source/tests/st/construct_input/
Dtest_outermost_input.py64 tensor_w = Tensor(np.ones((2, 2), np.float32) * 4) variable