Searched refs:GetSpanForData (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes_test.cc | 1332 inline absl::Span<const T> GetSpanForData(const InputOutputData& data) { in GetSpanForData() function
1339 auto span = GetSpanForData<float>(data); in GetDataAsFloat()
1344 GetSpanForData<Eigen::half>(data)); in GetDataAsFloat()
1347 return CastTestVector<int32, float>(GetSpanForData<int32>(data)); in GetDataAsFloat()
2415 EXPECT_THAT(GetSpanForData<float>(output_data[0]), in TEST_F()
2520 EXPECT_THAT(GetSpanForData<float>(output_data[0]), ElementsAre(1, 3)); in TestMatMulHelper()
2522 EXPECT_THAT(GetSpanForData<float>(output_data[0]), ElementsAre(2, 3)); in TestMatMulHelper()
2547 EXPECT_THAT(GetSpanForData<float>(output_data[0]), ElementsAre(1, 3)); in TestMatMulHelper()
2549 EXPECT_THAT(GetSpanForData<float>(output_data[0]), ElementsAre(2, 3)); in TestMatMulHelper()
2690 EXPECT_THAT(GetSpanForData<float>(output_data[0]), in TEST_F()
[all …]
|