Searched defs:BatchTask (Results 1 – 1 of 1) sorted by relevance
63 struct BatchTask : public tensorflow::serving::BatchTask { struct65 int64 guid;67 Context propagated_context;69 std::vector<Tensor> inputs;70 std::vector<Tensor> captured_inputs;71 OpKernelContext* context;72 AsyncOpKernel::DoneCallback done_callback;98 std::unique_ptr<BatchTask> CreateSplitTask( argument102 virtual std::unique_ptr<BatchTask> CreateDerivedTask() { in CreateDerivedTask() argument