Searched refs:constant_tensor (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/tools/benchmark/ |
| D | benchmark_model_test.cc | 39 Tensor constant_tensor(DT_FLOAT, constant_shape); in CreateTestGraph() local
40 test::FillFn<float>(&constant_tensor, [](int) -> float { return 3.0; }); in CreateTestGraph()
45 auto m = ops::MatMul(root, placeholder, constant_tensor); in CreateTestGraph()
|
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| D | quantize_nodes_test.cc | 398 Tensor constant_tensor(DT_FLOAT, TensorShape({1, 2, 6, 1})); in TestQuantizeRelu() local
399 test::FillValues<float>(&constant_tensor, in TestQuantizeRelu()
402 Input::Initializer(constant_tensor)); in TestQuantizeRelu()
416 Tensor constant_tensor(DT_FLOAT, TensorShape({1, 2, 6, 1})); in TestQuantizeRelu6() local
417 test::FillValues<float>(&constant_tensor, in TestQuantizeRelu6()
420 Input::Initializer(constant_tensor)); in TestQuantizeRelu6()
434 Tensor constant_tensor(DT_FLOAT, TensorShape({1, 2, 6, 1})); in TestQuantizeMaxPool() local
435 test::FillValues<float>(&constant_tensor, in TestQuantizeMaxPool()
438 Input::Initializer(constant_tensor)); in TestQuantizeMaxPool()
453 Tensor constant_tensor(DT_FLOAT, TensorShape({1, 2, 6, 1})); in TestQuantizeAvgPool() local
[all …]
|