Searched refs:GetSpanForData (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes_test.cc | 1060 inline absl::Span<const T> GetSpanForData(const InputOutputData& data) { in GetSpanForData() function
1484 EXPECT_THAT(GetSpanForData<float>(output_data[0]), in TEST_F()
1571 EXPECT_THAT(GetSpanForData<float>(output_data[0]), in TEST_F()
1636 EXPECT_THAT(GetSpanForData<float>(output_data[0]), ElementsAre(1, 3)); in TEST_F()
1638 EXPECT_THAT(GetSpanForData<float>(output_data[0]), ElementsAre(2, 3)); in TEST_F()
1699 EXPECT_THAT(GetSpanForData<CType>(output_data[0]), in TestConvertBiasAdd()
1702 EXPECT_THAT(GetSpanForData<CType>(output_data[0]), in TestConvertBiasAdd()
1707 EXPECT_THAT(GetSpanForData<CType>(output_data[0]), in TestConvertBiasAdd()
1711 EXPECT_THAT(GetSpanForData<CType>(output_data[0]), in TestConvertBiasAdd()
1798 EXPECT_THAT(GetSpanForData<CType>(output_data[0]), in TestBinaryTensorOpWeightNoBroadcast()
[all …]
|