Searched refs:range_shape (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | quantize_and_dequantize_op_test.cc | 143 const TensorShape range_shape = in TEST_P() local
146 AddInputFromArray<float>(range_shape, init_value); // Min in TEST_P()
147 AddInputFromArray<float>(range_shape, init_value); // Max in TEST_P()
194 const TensorShape range_shape = in TEST_P() local
197 AddInputFromArray<float>(range_shape, init_value); // Min in TEST_P()
198 AddInputFromArray<float>(range_shape, init_value); // Max in TEST_P()
248 const TensorShape range_shape = in TEST_P() local
251 AddInputFromArray<float>(range_shape, init_value); // Min in TEST_P()
252 AddInputFromArray<float>(range_shape, init_value); // Max in TEST_P()
331 const TensorShape range_shape = in TEST_P() local
[all …]
|
| D | quantize_and_dequantize_op.cc | 107 auto range_shape = (axis_ == -1) ? TensorShape({}) : TensorShape({depth}); in Compute() local
109 range_shape, &input_min_tensor)); in Compute()
111 range_shape, &input_max_tensor)); in Compute()
266 auto range_shape = (axis_ == -1) ? TensorShape({}) : TensorShape({depth}); in Compute() local
268 range_shape, &input_min_tensor)); in Compute()
270 range_shape, &input_max_tensor)); in Compute()
|