Searched refs:tfl_tensor (Results 1 – 2 of 2) sorted by relevance
100 const TfLiteTensor* tfl_tensor; in GetOpSignatureTensorSpecs() local102 tfl_tensor = &context->tensors[tensor_no]; in GetOpSignatureTensorSpecs()104 tfl_tensor = context->GetTensor(context, tensor_no); in GetOpSignatureTensorSpecs()106 if (tfl_tensor != nullptr) { in GetOpSignatureTensorSpecs()107 tensor_spec.type = tfl_tensor->type; in GetOpSignatureTensorSpecs()108 tensor_spec.is_const = (tfl_tensor->allocation_type == kTfLiteMmapRo); in GetOpSignatureTensorSpecs()109 if (tfl_tensor->dims) { in GetOpSignatureTensorSpecs()110 for (int32_t j = 0; j < tfl_tensor->dims->size; ++j) { in GetOpSignatureTensorSpecs()111 tensor_spec.dims.push_back(tfl_tensor->dims->data[j]); in GetOpSignatureTensorSpecs()114 tensor_spec.is_shape_dynamic = HasUnspecifiedDimension(tfl_tensor); in GetOpSignatureTensorSpecs()
633 TfLiteTensor* tfl_tensor = &context->tensors[input_tensor_index]; in ValidateOutputTensorShapeConsistency() local636 if (IsConstantTensor(tfl_tensor)) { in ValidateOutputTensorShapeConsistency()640 const auto dims_array = tfl_tensor->dims; in ValidateOutputTensorShapeConsistency()674 TfLiteTensor* tfl_tensor = &context->tensors[output_tensor_index]; in ValidateOutputTensorShapeConsistency() local679 GetShapeDebugString(tfl_tensor->dims); in ValidateOutputTensorShapeConsistency()686 (tfl_tensor->dims->size > 0)) { in ValidateOutputTensorShapeConsistency()