Searched refs:float_tensor (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/lite/tools/optimize/ |
| D | quantize_weights_test.cc | 142 const auto float_tensor = float_graph->tensors()->Get(i); in TEST_F() local
144 EXPECT_EQ(quant_tensor->buffer(), float_tensor->buffer()); in TEST_F()
145 EXPECT_EQ(quant_tensor->is_variable(), float_tensor->is_variable()); in TEST_F()
147 GetAsVector(float_tensor->shape())); in TEST_F()
148 EXPECT_EQ(quant_tensor->name()->str(), float_tensor->name()->str()); in TEST_F()
149 EXPECT_EQ(quant_tensor->type(), float_tensor->type()); in TEST_F()
180 const auto float_tensor = float_graph->tensors()->Get(i); in TEST_F() local
181 EXPECT_EQ(quant_tensor->buffer(), float_tensor->buffer()); in TEST_F()
182 EXPECT_EQ(quant_tensor->is_variable(), float_tensor->is_variable()); in TEST_F()
184 GetAsVector(float_tensor->shape())); in TEST_F()
[all …]
|
| D | quantize_model_test.cc | 84 const auto float_tensor = float_graph->tensors()->Get(i); in TEST_F() local
85 EXPECT_EQ(quant_tensor->buffer, float_tensor->buffer()); in TEST_F()
86 EXPECT_EQ(quant_tensor->is_variable, float_tensor->is_variable()); in TEST_F()
87 EXPECT_EQ(quant_tensor->shape, GetAsVector(float_tensor->shape())); in TEST_F()
88 EXPECT_EQ(quant_tensor->name, float_tensor->name()->str()); in TEST_F()
|
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| D | fuse_quantized_convolution.cc | 180 Tensor float_tensor; in FuseQuantizedConvolutionAndRequantize() local
181 if(!float_tensor.FromProto(float_tensor_proto)) { in FuseQuantizedConvolutionAndRequantize()
185 if (float_tensor.dtype() != DT_FLOAT) { in FuseQuantizedConvolutionAndRequantize()
189 float *p_bias_float = float_tensor.flat<float>().data(); in FuseQuantizedConvolutionAndRequantize()
191 Tensor int32_tensor = Tensor(DT_QINT32, float_tensor.shape()); in FuseQuantizedConvolutionAndRequantize()
197 int64 nelems = float_tensor.NumElements(); in FuseQuantizedConvolutionAndRequantize()
|
| D | quantize_nodes_test.cc | 129 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/lite/testing/model_coverage/ |
| D | 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/ |
| D | remote_fused_graph_execute_utils.cc | 226 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/ |
| D | feature_column_test.py | 777 float_tensor = sparse_tensor.SparseTensor(
784 'a_float': float_tensor
|
| D | feature_column_v2_test.py | 943 float_tensor = sparse_tensor.SparseTensor(
948 'a_float': float_tensor
|