HomeSort by relevance Sort by last modified time
    Searched refs:CopyNumpyArrayToLiteral (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/python/
numpy_bridge.h 90 Status CopyNumpyArrayToLiteral(int np_type, PyArrayObject* py_array,
97 void CopyNumpyArrayToLiteral(PyArrayObject* py_array, Literal* literal) {
numpy_bridge.cc 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)
    [all...]

Completed in 127 milliseconds