Lines Matching refs:tensorflow
1 /* Copyright 2017 The TensorFlow Authors. All Rights Reserved.
19 #include "tensorflow/compiler/xla/xla.pb.h"
20 #include "tensorflow/core/lib/core/status.h"
34 virtual tensorflow::Status TransferToClient(
37 virtual tensorflow::Status TransferToServer(
40 virtual tensorflow::Status TransferToInfeed(
43 virtual tensorflow::Status TransferFromOutfeed(
47 virtual tensorflow::Status ResetDevice(const ResetDeviceRequest* arg,
50 virtual tensorflow::Status LoadComputationSnapshot(
54 virtual tensorflow::Status Execute(const ExecuteRequest* arg,
57 virtual tensorflow::Status ExecuteParallel(
60 virtual tensorflow::Status ExecuteAsync(const ExecuteAsyncRequest* arg,
63 virtual tensorflow::Status WaitForExecution(
66 virtual tensorflow::Status DeconstructTuple(
69 virtual tensorflow::Status GetComputationStats(
72 virtual tensorflow::Status GetComputationShape(
76 virtual tensorflow::Status GetShape(const GetShapeRequest* arg,
79 virtual tensorflow::Status CreateChannelHandle(
83 virtual tensorflow::Status GetDeviceHandles(
87 virtual tensorflow::Status Computation(const ComputationRequest* arg,
90 virtual tensorflow::Status Op(const OpRequest* arg, OpResponse* result) = 0;
92 virtual tensorflow::Status GetLocalShape(const GetLocalShapeRequest* arg,
95 virtual tensorflow::Status SetReturnValue(
98 virtual tensorflow::Status IsConstant(const IsConstantRequest* arg,
101 virtual tensorflow::Status ComputeConstant(
105 virtual tensorflow::Status SnapshotComputation(
110 virtual tensorflow::Status Unregister(const UnregisterRequest* arg,
