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

  /external/tensorflow/tensorflow/compiler/xla/service/
user_computation.h 39 // A UserComputation is the built-up computation that users create via the
49 class UserComputation {
57 static StatusOr<std::unique_ptr<UserComputation>> MakeWithRemapping(
63 explicit UserComputation(const string& name, const ComputationHandle& handle);
126 const UserComputation& to_apply_computation);
159 const UserComputation& to_apply_computation);
208 const UserComputation& to_apply_computation);
213 const UserComputation& to_apply_computation);
219 const UserComputation& select_computation,
220 const UserComputation& scatter_computation)
    [all...]
user_computation.cc 156 /* static */ StatusOr<std::unique_ptr<UserComputation>>
157 UserComputation::MakeWithRemapping(
162 MakeUnique<UserComputation>(session_computation.name(), handle);
181 UserComputation::UserComputation(const string& name,
187 VLOG(1) << "New UserComputation \"" << name
191 ComputationDataHandle UserComputation::CreateComputationDataHandle() {
200 StatusOr<ComputationDataHandle> UserComputation::AddParameterInstruction(
229 Status UserComputation::AddSendInstruction(const SendRequest& send_request) {
250 StatusOr<ComputationDataHandle> UserComputation::AddRecvInstruction
    [all...]

Completed in 483 milliseconds