Lines Matching refs:py_array
386 PyArrayObject* py_array = reinterpret_cast<PyArrayObject*>(o);
387 int rank = PyArray_NDIM(py_array);
390 dimensions[i] = PyArray_DIM(py_array, i);
392 int np_type = PyArray_TYPE(py_array);
396 CopyNumpyArrayToLiteral(np_type, py_array, literal.get()));
404 Status CopyNumpyArrayToLiteral(int np_type, PyArrayObject* py_array,
408 CopyNumpyArrayToLiteral<bool>(py_array, literal);
411 CopyNumpyArrayToLiteral<int32>(py_array, literal);
414 CopyNumpyArrayToLiteral<int64>(py_array, literal);
417 CopyNumpyArrayToLiteral<uint8>(py_array, literal);
420 CopyNumpyArrayToLiteral<uint32>(py_array, literal);
423 CopyNumpyArrayToLiteral<uint64>(py_array, literal);
426 CopyNumpyArrayToLiteral<half>(py_array, literal);
429 CopyNumpyArrayToLiteral<float>(py_array, literal);
432 CopyNumpyArrayToLiteral<double>(py_array, literal);
442 PyArrayObject* py_array) {
445 CopyLiteralToNumpyArray<bool>(literal, py_array);
448 CopyLiteralToNumpyArray<int32>(literal, py_array);
451 CopyLiteralToNumpyArray<int64>(literal, py_array);
454 CopyLiteralToNumpyArray<uint8>(literal, py_array);
457 CopyLiteralToNumpyArray<uint32>(literal, py_array);
460 CopyLiteralToNumpyArray<uint64>(literal, py_array);
463 CopyLiteralToNumpyArray<half>(literal, py_array);
466 CopyLiteralToNumpyArray<float>(literal, py_array);
469 CopyLiteralToNumpyArray<double>(literal, py_array);
