Home
last modified time | relevance | path

Searched refs:float_tensor (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/lite/tools/optimize/
Dquantize_weights_test.cc166 const auto float_tensor = float_graph->tensors()->Get(i); in TEST_F() local
168 EXPECT_EQ(quant_tensor->buffer(), float_tensor->buffer()); in TEST_F()
169 EXPECT_EQ(quant_tensor->is_variable(), float_tensor->is_variable()); in TEST_F()
171 GetAsVector(float_tensor->shape())); in TEST_F()
172 EXPECT_EQ(quant_tensor->name()->str(), float_tensor->name()->str()); in TEST_F()
173 EXPECT_EQ(quant_tensor->type(), float_tensor->type()); in TEST_F()
204 const auto float_tensor = float_graph->tensors()->Get(i); in TEST_F() local
205 EXPECT_EQ(quant_tensor->buffer(), float_tensor->buffer()); in TEST_F()
206 EXPECT_EQ(quant_tensor->is_variable(), float_tensor->is_variable()); in TEST_F()
208 GetAsVector(float_tensor->shape())); in TEST_F()
[all …]
Dmodify_model_interface.cc204 TensorT* float_tensor = subgraph->tensors[tot.input_index].get(); in SetInputTypeToUINT8() local
205 float_tensor->type = TensorType_UINT8; in SetInputTypeToUINT8()
206 if (float_tensor->quantization == nullptr) { in SetInputTypeToUINT8()
207 float_tensor->quantization = absl::make_unique<QuantizationParametersT>(); in SetInputTypeToUINT8()
209 float_tensor->quantization->scale.push_back(quant_tensor_scale); in SetInputTypeToUINT8()
210 float_tensor->quantization->zero_point.push_back(quant_tensor_zp + 128); in SetInputTypeToUINT8()
231 TensorT* float_tensor = subgraph->tensors[tot.output_index].get(); in SetOutputTypeToUINT8() local
232 float_tensor->type = TensorType_UINT8; in SetOutputTypeToUINT8()
233 if (float_tensor->quantization == nullptr) { in SetOutputTypeToUINT8()
234 float_tensor->quantization = absl::make_unique<QuantizationParametersT>(); in SetOutputTypeToUINT8()
[all …]
Dquantize_model_test.cc163 const auto float_tensor = float_graph->tensors()->Get(i); in TEST_P() local
164 EXPECT_EQ(quant_tensor->buffer, float_tensor->buffer()); in TEST_P()
165 EXPECT_EQ(quant_tensor->is_variable, float_tensor->is_variable()); in TEST_P()
166 EXPECT_EQ(quant_tensor->shape, GetAsVector(float_tensor->shape())); in TEST_P()
167 EXPECT_EQ(quant_tensor->name, float_tensor->name()->str()); in TEST_P()
168 EXPECT_EQ(quant_tensor->type, float_tensor->type()); in TEST_P()
186 const auto float_tensor = float_graph->tensors()->Get(i); in TEST_P() local
187 EXPECT_EQ(quant_tensor->buffer, float_tensor->buffer()); in TEST_P()
188 EXPECT_EQ(quant_tensor->is_variable, float_tensor->is_variable()); in TEST_P()
189 EXPECT_EQ(quant_tensor->shape, GetAsVector(float_tensor->shape())); in TEST_P()
[all …]
/external/tensorflow/tensorflow/lite/testing/model_coverage/
Dmodel_coverage_lib.py644 float_tensor["dtype"] == quant_tensors_map[float_tensor["name"]]["dtype"]
645 for float_tensor in float_tensors)
844 float_tensor["dtype"] == quant_tensors_map[float_tensor["name"]]["dtype"]
845 for float_tensor in float_tensors)
/external/tensorflow/tensorflow/lite/python/
Dutil.py671 float_tensor, quant_tensor = tensors[op.inputs[0]], tensors[op.outputs[0]]
673 float_type = _convert_tflite_enum_type_to_tf_type(float_tensor.type)
681 float_tensor.name, get_tf_type_name(float_type)))
766 quant_tensor, float_tensor = tensors[op.inputs[0]], tensors[op.outputs[0]]
767 float_type = _convert_tflite_enum_type_to_tf_type(float_tensor.type)
775 float_tensor.name, get_tf_type_name(float_type)))
/external/tensorflow/tensorflow/core/common_runtime/eager/
Dcustom_device_test.cc132 Tensor float_tensor(DT_FLOAT, {}); in TEST() local
134 TensorHandle::CreateLocalHandle(std::move(float_tensor), physical_device, in TEST()
/external/tensorflow/tensorflow/tools/graph_transforms/
Dquantize_nodes_test.cc129 const Tensor& float_tensor = float_input.second; in TestGraphWithInputRange() local
130 Tensor quantized_tensor(DT_QUINT8, float_tensor.shape()); in TestGraphWithInputRange()
131 FloatTensorToQuantizedInPlace<quint8>(float_tensor, range_min, range_max, in TestGraphWithInputRange()
/external/tensorflow/tensorflow/core/kernels/
Dremote_fused_graph_execute_utils.cc226 auto float_tensor = input_tensor.flat<float>(); in DryRunInference() local
227 float_tensor = float_tensor.constant(0.0f); in DryRunInference()
/external/tensorflow/tensorflow/python/feature_column/
Dfeature_column_v2_test.py886 float_tensor = sparse_tensor.SparseTensor(
891 'a_float': float_tensor
Dfeature_column_test.py761 float_tensor = sparse_tensor.SparseTensor(
768 'a_float': float_tensor