Searched refs:TensorProtoHelper (Results 1 – 3 of 3) sorted by relevance
462 EXPECT_EQ(tensor::internal::TensorProtoHelper<float>::NumValues(tensor_proto), in TEST()468 EXPECT_EQ(tensor::internal::TensorProtoHelper<int>::NumValues(tensor_proto), in TEST()474 EXPECT_EQ(tensor::internal::TensorProtoHelper<uint8>::NumValues(tensor_proto), in TEST()479 EXPECT_EQ(tensor::internal::TensorProtoHelper<bool>::NumValues(tensor_proto), in TEST()486 tensor::internal::TensorProtoHelper<Eigen::half>::NumValues(tensor_proto), in TEST()492 EXPECT_EQ(tensor::internal::TensorProtoHelper<std::complex<float>>::NumValues( in TEST()538 using TensorProtoHelper = tensor::internal::TensorProtoHelper<T>; in ConstantTailTest() typedef539 using FieldType = typename TensorProtoHelper::FieldType; in ConstantTailTest()560 EXPECT_EQ(TensorProtoHelper::NumValues(tensor_proto), 0); in ConstantTailTest()563 EXPECT_LE(TensorProtoHelper::NumValues(tensor_proto), in ConstantTailTest()
188 using TypeHelper = internal::TensorProtoHelper<T>; in CompressTensorContent()189 using FieldType = typename internal::TensorProtoHelper<T>::FieldType; in CompressTensorContent()267 using TypeHelper = internal::TensorProtoHelper<T>; in CompressRepeatedField()268 using FieldType = typename internal::TensorProtoHelper<T>::FieldType; in CompressRepeatedField()
195 class TensorProtoHelper : public std::true_type {250 class TensorProtoHelper<string> : public std::true_type {275 typename std::enable_if<internal::TensorProtoHelper<Type>::value,287 using TypeHelper = internal::TensorProtoHelper<Type>;