Searched refs:old_tensor (Results 1 – 2 of 2) sorted by relevance
56 Tensor old_tensor; in QuantizeWeights() local57 if (!old_tensor.FromProto(old_const_node.attr().at("value").tensor())) { in QuantizeWeights()61 const size_t num_elements = old_tensor.NumElements(); in QuantizeWeights()68 const float* old_values = old_tensor.flat<float>().data(); in QuantizeWeights()95 Tensor quantized_tensor(DT_QUINT8, old_tensor.shape()); in QuantizeWeights()96 FloatTensorToQuantizedInPlace<quint8>(old_tensor, min, max, in QuantizeWeights()
53 Tensor old_tensor; in RoundWeights() local54 if (!old_tensor.FromProto(old_const_node.attr().at("value").tensor())) { in RoundWeights()58 const size_t num_elements = old_tensor.NumElements(); in RoundWeights()67 const float* old_values = old_tensor.flat<float>().data(); in RoundWeights()90 Tensor rounded_tensor(DT_FLOAT, old_tensor.shape()); in RoundWeights()