Searched refs:first_partition_input (Results 1 – 1 of 1) sorted by relevance
61 const TfLiteTensor& first_partition_input, in GetFirstDimensionSizeT() argument67 return *GetTensorData<INDEX_TYPE>(&first_partition_input); in GetFirstDimensionSizeT()73 const auto shape = GetTensorShape(&first_partition_input); in GetFirstDimensionSizeT()86 const TfLiteTensor& first_partition_input, in GetFirstDimensionSize() argument88 switch (first_partition_input.type) { in GetFirstDimensionSize()90 return GetFirstDimensionSizeT<int32_t>(context, first_partition_input, in GetFirstDimensionSize()93 return GetFirstDimensionSizeT<int64_t>(context, first_partition_input, in GetFirstDimensionSize()621 TfLiteTensor& first_partition_input = in Eval() local626 GetFirstDimensionSize(context, first_partition_input, attributes); in Eval()