Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/delegates/flex/
Dkernel.cc375 DelegateKernel::DelegateKernel() : op_data_(new OpData) {} in DelegateKernel()
382 op_data_->eager_context = flex_delegate_data->GetEagerContext(); in Init()
383 op_data_->buffer_map = flex_delegate_data->GetBufferMap(context); in Init()
388 op_data_->subgraph_outputs.push_back(tensor_index); in Init()
394 op_data_->subgraph_inputs.push_back(tensor_index); in Init()
397 op_data_->nodes.reserve(params->nodes_to_replace->size); in Init()
406 op_data_->nodes.emplace_back(new OpNode(node->inputs, node->outputs)); in Init()
407 OpNode& node_data = *op_data_->nodes.back(); in Init()
415 status = node_data.BuildEagerOp(op_data_->eager_context); in Init()
427 for (auto& node_data : op_data_->nodes) { in Init()
[all …]
Dkernel.h43 std::unique_ptr<OpData> op_data_; variable