Searched defs:inp_dims (Results 1 – 3 of 3) sorted by relevance
178 std::vector<int64_t> inp_dims; in Compile() local
3934 def _test_bfloat16_ops(test_case, op, device, inp_dims=(), prec=1e-2, scale_factor=None): argument
11398 def test_helper(fn, device, inp_dims, prec=None): argument