Home
last modified time | relevance | path

Searched defs:expected_output_dims (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes_test.cc1735 std::vector<int> expected_output_dims; member
1869 const std::vector<std::vector<int>>& expected_output_dims, in BuildAndRun()
1912 const std::vector<std::vector<int>>& expected_output_dims, in TestOpConverterMultiOut()
1928 const std::vector<int>& expected_output_dims, in TestOpConverter()
2404 const std::vector<int> expected_output_dims(shape.begin() + 1, shape.end()); in TEST_F() local
3168 const std::vector<std::vector<int32>> expected_output_dims; in TEST_P() member
3733 std::vector<int> expected_output_dims; in TEST_F() member
4298 std::vector<int> expected_output_dims; in TEST_F() member
4467 std::vector<int> expected_output_dims; in TEST_P() member
4607 std::vector<int> expected_output_dims; in TEST_F() member
[all …]
/external/tensorflow/tensorflow/lite/delegates/xnnpack/
Dxnnpack_delegate.cc1993 const int expected_output_dims = reducer_params->keep_dims ? 4 : 2; in VisitMeanNode() local