Searched refs:TensorProtoHelper (Results 1 – 3 of 3) sorted by relevance
463 EXPECT_EQ(tensor::internal::TensorProtoHelper<float>::NumValues(tensor_proto), in TEST()469 EXPECT_EQ(tensor::internal::TensorProtoHelper<int>::NumValues(tensor_proto), in TEST()475 EXPECT_EQ(tensor::internal::TensorProtoHelper<uint8>::NumValues(tensor_proto), in TEST()480 EXPECT_EQ(tensor::internal::TensorProtoHelper<bool>::NumValues(tensor_proto), in TEST()487 tensor::internal::TensorProtoHelper<Eigen::half>::NumValues(tensor_proto), in TEST()493 EXPECT_EQ(tensor::internal::TensorProtoHelper<std::complex<float>>::NumValues( in TEST()553 using TensorProtoHelper = tensor::internal::TensorProtoHelper<T>; in ConstantTailTest() typedef554 using FieldType = typename TensorProtoHelper::FieldType; in ConstantTailTest()575 EXPECT_EQ(TensorProtoHelper::NumValues(tensor_proto), 0); in ConstantTailTest()578 EXPECT_LE(TensorProtoHelper::NumValues(tensor_proto), in ConstantTailTest()
197 class TensorProtoHelper : public std::true_type {252 class TensorProtoHelper<string> : public std::true_type {277 typename std::enable_if<internal::TensorProtoHelper<Type>::value,289 using TypeHelper = internal::TensorProtoHelper<Type>;
195 using TypeHelper = internal::TensorProtoHelper<T>; in CompressTensorContent()196 using FieldType = typename internal::TensorProtoHelper<T>::FieldType; in CompressTensorContent()274 using TypeHelper = internal::TensorProtoHelper<T>; in CompressRepeatedField()275 using FieldType = typename internal::TensorProtoHelper<T>::FieldType; in CompressRepeatedField()