Searched refs:temp_tensor (Results 1 – 3 of 3) sorted by relevance
210 Tensor temp_tensor; in SolveWithGtsv() local213 context->allocate_temp(DT_UINT8, temp_shape, &temp_tensor)); in SolveWithGtsv()214 void* buffer = temp_tensor.flat<std::uint8_t>().data(); in SolveWithGtsv()318 Tensor temp_tensor; in ComputeWithGtsvBatched() local321 context->allocate_temp(DT_UINT8, temp_shape, &temp_tensor)); in ComputeWithGtsvBatched()322 void* buffer = temp_tensor.flat<std::uint8_t>().data(); in ComputeWithGtsvBatched()
320 TfLiteTensor* temp_tensor = &context->tensors[op_data->temp_tensor_index];321 temp_tensor->type = kTfLiteFloat32;322 temp_tensor->allocation_type = kTfLiteArenaRw;
1383 TensorProto temp_tensor; in GetTensorShapeProtoFromTensorProto() local1384 tensor.AsProtoField(&temp_tensor); in GetTensorShapeProtoFromTensorProto()1388 for (const auto& value : temp_tensor.type##_val()) { \ in GetTensorShapeProtoFromTensorProto()