Home
last modified time | relevance | path

Searched refs:scalar_values (Results 1 – 2 of 2) sorted by relevance

/external/pytorch/torch/csrc/jit/mobile/nnc/
Dcontext.cpp264 std::vector<int64_t> scalar_values; in run() local
275 scalar_values.reserve(sym_shape_positions_.size()); in run()
280 scalar_values.push_back(dim); in run()
281 args[i + offset] = &scalar_values[scalar_values.size() - 1]; in run()
/external/pytorch/test/inductor/
Dtest_torchinductor.py962 scalar_values = [1.0, 2.0, 3.0]
964 for scalar_value in scalar_values:
972 for scalar_value in scalar_values: