| /external/tensorflow/tensorflow/lite/tools/optimize/ |
| quantize_weights_test.cc | 142 const auto float_tensor = float_graph->tensors()->Get(i); local
144 EXPECT_EQ(quant_tensor->buffer(), float_tensor->buffer());
145 EXPECT_EQ(quant_tensor->is_variable(), float_tensor->is_variable());
147 GetAsVector(float_tensor->shape()));
148 EXPECT_EQ(quant_tensor->name()->str(), float_tensor->name()->str());
149 EXPECT_EQ(quant_tensor->type(), float_tensor->type());
180 const auto float_tensor = float_graph->tensors()->Get(i); local
181 EXPECT_EQ(quant_tensor->buffer(), float_tensor->buffer());
182 EXPECT_EQ(quant_tensor->is_variable(), float_tensor->is_variable());
184 GetAsVector(float_tensor->shape()))
[all...] |
| quantize_model_test.cc | 84 const auto float_tensor = float_graph->tensors()->Get(i); local
85 EXPECT_EQ(quant_tensor->buffer, float_tensor->buffer());
86 EXPECT_EQ(quant_tensor->is_variable, float_tensor->is_variable());
87 EXPECT_EQ(quant_tensor->shape, GetAsVector(float_tensor->shape()));
88 EXPECT_EQ(quant_tensor->name, float_tensor->name()->str());
|
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| fuse_quantized_convolution.cc | 180 Tensor float_tensor; local
181 if(!float_tensor.FromProto(float_tensor_proto)) {
185 if (float_tensor.dtype() != DT_FLOAT) {
189 float *p_bias_float = float_tensor.flat<float>().data();
191 Tensor int32_tensor = Tensor(DT_QINT32, float_tensor.shape());
197 int64 nelems = float_tensor.NumElements();
|
| quantize_nodes_test.cc | 129 const Tensor& float_tensor = float_input.second; local
130 Tensor quantized_tensor(DT_QUINT8, float_tensor.shape());
131 FloatTensorToQuantizedInPlace<quint8>(float_tensor, range_min, range_max,
[all...] |
| /external/tensorflow/tensorflow/lite/testing/model_coverage/ |
| model_coverage_lib.py | 304 float_tensor["dtype"] == quant_tensors_map[float_tensor["name"]]["dtype"]
305 for float_tensor in float_tensors)
|
| /external/tensorflow/tensorflow/core/kernels/ |
| remote_fused_graph_execute_utils.cc | 226 auto float_tensor = input_tensor.flat<float>(); local
227 float_tensor = float_tensor.constant(0.0f);
[all...] |
| /external/tensorflow/tensorflow/python/feature_column/ |
| feature_column_test.py | 777 float_tensor = sparse_tensor.SparseTensor(
784 'a_float': float_tensor
[all...] |
| feature_column_v2_test.py | [all...] |
