Searched defs:BatchTask (Results 1 – 1 of 1) sorted by relevance
282 struct BatchTask : public serving::BatchTask { struct in tensorflow::BatchResource284 int64 guid;286 std::vector<Tensor> inputs;287 std::vector<Tensor> captured_inputs;288 OpKernelContext* context;289 AsyncOpKernel::DoneCallback done_callback;291 size_t size() const override { return inputs[0].shape().dim_size(0); } in size()