Home
last modified time | relevance | path

Searched refs:batch_components (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/batching_util/
Dbatch_resource_base.cc238 std::unique_ptr<BatchTask> batch_components; in RegisterInput() local
239 TF_RETURN_IF_ERROR(CreateBatchTask(context, &batch_components)); in RegisterInput()
240 batch_components->start_time = EnvTime::NowNanos(); in RegisterInput()
241 batch_components->guid = guid; in RegisterInput()
242 batch_components->propagated_context = Context(ContextKind::kThread); in RegisterInput()
245 batch_components->inputs.reserve(tensors.size()); in RegisterInput()
257 batch_components->inputs.push_back(tensor); in RegisterInput()
306 batch_components->captured_inputs.reserve(captured_tensors.size()); in RegisterInput()
308 batch_components->captured_inputs.push_back(captured_tensor); in RegisterInput()
311 batch_components->context = context; in RegisterInput()
[all …]