Searched refs:output_view (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | auto_mixed_precision_test.cc | 190 GraphView output_view(&output); in TestSimpleUnaryInferOp() local
191 EXPECT_EQ(output_view.GetNode("input")->attr().at("dtype").type(), in TestSimpleUnaryInferOp()
193 EXPECT_EQ(output_view.GetNode("allow1")->attr().at("T").type(), DT_HALF); in TestSimpleUnaryInferOp()
194 EXPECT_EQ(output_view.GetNode("infer1")->attr().at("T").type(), DT_HALF); in TestSimpleUnaryInferOp()
195 EXPECT_EQ(output_view.GetNode("allow2")->attr().at("T").type(), DT_HALF); in TestSimpleUnaryInferOp()
231 GraphView output_view(&output); in TEST_F() local
232 EXPECT_EQ(output_view.GetNode("input")->attr().at("dtype").type(), DT_FLOAT); in TEST_F()
233 EXPECT_EQ(output_view.GetNode("deny1")->attr().at("T").type(), DT_FLOAT); in TEST_F()
234 EXPECT_EQ(output_view.GetNode("clr1")->attr().at("T").type(), DT_FLOAT); in TEST_F()
235 EXPECT_EQ(output_view.GetNode("infer1")->attr().at("T").type(), DT_FLOAT); in TEST_F()
[all …]
|
| /external/libtextclassifier/native/utils/ |
| D | tflite-model-executor.h | 138 TensorView<T> output_view = OutputView<T>(output_index, interpreter); in Output() local
139 return std::vector<T>(output_view.data(), in Output()
140 output_view.data() + output_view.size()); in Output()
|