Searched refs:SetAval (Results 1 – 4 of 4) sorted by relevance
109 void SetAval(pybind11::object aval);
335 void PyBuffer::SetAval(pybind11::object aval) { in SetAval() function in xla::PyBuffer
279 .def_property("aval", &PyBuffer::GetAval, &PyBuffer::SetAval) in PYBIND11_MODULE()
1191 buffer->SetAval(out_avals[i]); in Call()1251 buffer->SetAval((*shaped_array)( in BuildJaxjitSubmodule()