Searched refs:IntSpanToTuple (Results 1 – 6 of 6) sorted by relevance
253 py::tuple IntSpanToTuple(absl::Span<int64 const> xs) { in IntSpanToTuple() function256 py::tuple IntSpanToTuple(absl::Span<int const> xs) { in IntSpanToTuple() function
59 return IntSpanToTuple(buffer()->on_device_shape().dimensions()); in python_shape()185 result["shape"] = IntSpanToTuple(buffer_->on_device_shape().dimensions()); in CudaArrayInterface()
171 return IntSpanToTuple(device.coords()); in PYBIND11_MODULE()287 return IntSpanToTuple( in PYBIND11_MODULE()
87 pybind11::tuple IntSpanToTuple(absl::Span<int64 const> xs);88 pybind11::tuple IntSpanToTuple(absl::Span<int const> xs);
205 return IntSpanToTuple(shape.dimensions()); in BuildXlaCompilerSubmodule()
1269 return xla::IntSpanToTuple(sig.shape); in BuildJaxjitSubmodule()