Searched refs:PjRtStreamExecutorExecutable (Results 1 – 3 of 3) sorted by relevance
1566 PjRtStreamExecutorExecutable::PjRtStreamExecutorExecutable( in PjRtStreamExecutorExecutable() function in xla::PjRtStreamExecutorExecutable1608 Status PjRtStreamExecutorExecutable::SetUpDonation(bool tuple_inputs) { in SetUpDonation()1620 absl::string_view PjRtStreamExecutorExecutable::name() const { in name()1629 bool PjRtStreamExecutorExecutable::MustDonateParameter(int executable_idx, in MustDonateParameter()1635 PjRtStreamExecutorExecutable::MakeExecutionInputsAndWaitForEvents( in MakeExecutionInputsAndWaitForEvents()1681 StatusOr<ScopedShapedBuffer> PjRtStreamExecutorExecutable::EnqueueExecution( in EnqueueExecution()1822 PjRtStreamExecutorExecutable::MakeOutputBuffers( in MakeOutputBuffers()1857 PjRtStreamExecutorExecutable::ExecuteHelper( in ExecuteHelper()1942 PjRtStreamExecutorExecutable::Execute( in Execute()2057 PjRtStreamExecutorExecutable::ExecuteSharded( in ExecuteSharded()[all …]
601 class PjRtStreamExecutorExecutable : public PjRtExecutable {603 PjRtStreamExecutorExecutable(611 ~PjRtStreamExecutorExecutable() override = default;
129 tensorflow::down_cast<const PjRtStreamExecutorExecutable*>( in ExecutableFingerprint()