Searched refs:ValueOrThrow (Results 1 – 4 of 4) sorted by relevance
165 return absl::make_unique<Shape>(ValueOrThrow(ParseShape(s))); in BuildXlaCompilerSubmodule()191 PrimitiveType type = ValueOrThrow(DtypeToPrimitiveType(dtype)); in BuildXlaCompilerSubmodule()210 return ValueOrThrow(PrimitiveTypeToDtype(shape.element_type())); in BuildXlaCompilerSubmodule()217 return ValueOrThrow(PrimitiveTypeToDtype(shape.element_type())); in BuildXlaCompilerSubmodule()
367 return ValueOrThrow(client->BufferFromHostBuffer( in ConvertToScalarBuffer()623 ValueOrThrow(pyclient.pjrt_client()->BufferFromHostBuffer( in HandleComplex()632 ValueOrThrow(pyclient.pjrt_client()->BufferFromHostBuffer( in HandleComplex()656 ValueOrThrow(buffer->buffer()->CopyToDevice(to_device)); in HandleDeviceArray()667 ValueOrThrow(pyclient.PjRtBufferFromPyval( in HandleBufferFromPyval()691 ValueOrThrow(pyclient.PjRtBufferFromPyval( in HandleUint64()702 ValueOrThrow(pyclient.PjRtBufferFromPyval( in HandleUint64()731 ValueOrThrow(pyclient.PjRtBufferFromPyval( in HandleNdarray()1181 ValueOrThrow(cache_entry->executable->PjRtExecute(arguments.arg_buffers)); in Call()
327 ValueOrThrow(cache_entry->executable->ExecuteOnLocalDevices(arg_buffers)); in Call()
41 T ValueOrThrow(StatusOr<T> v) { in ValueOrThrow() function