Searched refs:type_vector (Results 1 – 2 of 2) sorted by relevance
657 context->GetAttr(attr_name, &type_vector)); in GetAttrKernel()693 DataTypeVector type_vector; member in tensorflow::__anon706e3cb20111::GetAttrKernel842 ASSERT_EQ(2, get_attr_kernel->type_vector.size()); in TEST_F()843 EXPECT_EQ(DT_INT32, get_attr_kernel->type_vector[0]); in TEST_F()844 EXPECT_EQ(DT_BOOL, get_attr_kernel->type_vector[1]); in TEST_F()
1089 absl::InlinedVector<unsigned int, 4> type_vector; in SetOpAttrValueScalar() local1090 type_vector.reserve(type_size); in SetOpAttrValueScalar()1092 type_vector.push_back(default_value.list().type(i)); in SetOpAttrValueScalar()1096 reinterpret_cast<const TF_DataType*>(type_vector.data()), in SetOpAttrValueScalar()