Searched defs:BatchTask (Results 1 – 1 of 1) sorted by relevance
65 struct BatchTask : public tensorflow::serving::BatchTask { struct67 int64_t guid;69 Context propagated_context;71 std::vector<Tensor> inputs;72 std::vector<Tensor> captured_inputs;73 OpKernelContext* context;74 AsyncOpKernel::DoneCallback done_callback;100 std::unique_ptr<BatchTask> CreateSplitTask( argument115 virtual std::unique_ptr<BatchTask> CreateDerivedTask() { in CreateDerivedTask() argument