Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/kernels/
Dif.cc85 TfLiteTensor* subgraph_input = subgraph->tensor(subgraph->inputs()[i]); in Prepare() local
86 TF_LITE_ENSURE_EQ(context, input->type, subgraph_input->type); in Prepare()
146 TfLiteTensor* subgraph_input = in Eval() local
148 TF_LITE_ENSURE_EQ(context, input->bytes, subgraph_input->bytes); in Eval()
149 memcpy(subgraph_input->data.raw, input->data.raw, input->bytes); in Eval()
/external/tensorflow/tensorflow/core/kernels/
Dremote_fused_graph_execute_utils.cc848 for (const string& subgraph_input : std::get<1>(cluster)) { in BuildClusterSubgraphDef() local
849 const TensorId tid = ParseTensorName(subgraph_input); in BuildClusterSubgraphDef()