Searched refs:PjRtStreamExecutorExecutable (Results 1 – 3 of 3) sorted by relevance
149 tensorflow::down_cast<const PjRtStreamExecutorExecutable*>( in ExecutableFingerprint()158 const PjRtStreamExecutorExecutable* se_executable = in SerializeExecutable()159 tensorflow::down_cast<const PjRtStreamExecutorExecutable*>(&executable); in SerializeExecutable()200 auto pjrt_executable = std::make_unique<PjRtStreamExecutorExecutable>( in DeserializeExecutable()
1761 PjRtStreamExecutorExecutable::PjRtStreamExecutorExecutable( in PjRtStreamExecutorExecutable() function in xla::PjRtStreamExecutorExecutable1815 Status PjRtStreamExecutorExecutable::SetUpDonation(bool tuple_inputs) { in SetUpDonation()1827 absl::string_view PjRtStreamExecutorExecutable::name() const { in name()1836 absl::Span<int const> PjRtStreamExecutorExecutable::ParametersThatMustBeDonated( in ParametersThatMustBeDonated()1842 PjRtStreamExecutorExecutable::MakeExecutionInputsAndWaitForEvents( in MakeExecutionInputsAndWaitForEvents()1899 StatusOr<ScopedShapedBuffer> PjRtStreamExecutorExecutable::EnqueueExecution( in EnqueueExecution()2052 PjRtStreamExecutorExecutable::MakeOutputBuffers( in MakeOutputBuffers()2091 PjRtStreamExecutorExecutable::ExecuteHelper( in ExecuteHelper()2195 PjRtStreamExecutorExecutable::Execute( in Execute()2320 PjRtStreamExecutorExecutable::ExecuteSharded( in ExecuteSharded()[all …]
710 class PjRtStreamExecutorExecutable : public PjRtLoadedExecutable {712 PjRtStreamExecutorExecutable(720 ~PjRtStreamExecutorExecutable() override = default;