Searched refs:concat_temp (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | lstm.cc | 2234 TfLiteTensor* concat_temp; in Prepare() local
2236 context, GetOutputSafe(context, node, kOutputConcatTemp, &concat_temp)); in Prepare()
2252 context, context->ResizeTensor(context, concat_temp, concat_temp_size)); in Prepare()
2288 TfLiteTensor* concat_temp; in Eval() local
2290 context, GetOutputSafe(context, node, kOutputConcatTemp, &concat_temp)); in Eval()
2300 concat_temp->type == kTfLiteFloat32 && in Eval()
2315 GetTensorShape(concat_temp), GetTensorData<float>(concat_temp), in Eval()
2324 concat_temp->type == kTfLiteUInt8 && in Eval()
2362 GetTensorShape(concat_temp), GetTensorData<uint8_t>(concat_temp), in Eval()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/ |
| D | model_builder.cc | 978 Value* concat_temp; in ParseBasic() local
981 reader->ReadValueByTensorIdx(concat_tensor_idx, &concat_temp)); in ParseBasic()
989 RETURN_IF_ERROR(graph->SetProducer(concat_node->id, concat_temp->id)); in ParseBasic()
991 RETURN_IF_ERROR(graph->AddConsumer(fc_node->id, concat_temp->id)); in ParseBasic()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/ir/ |
| D | tfl_ops.td | 3784 TFL_2DTensorOf<[F32, QUI8]>:$concat_temp,
|