Home
last modified time | relevance | path

Searched refs:type_vector (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/framework/
Dop_kernel_test.cc773 context->GetAttr(attr_name, &type_vector)); in GetAttrKernel()
809 DataTypeVector type_vector; member in tensorflow::__anonc25ee1c30111::GetAttrKernel
950 ASSERT_EQ(2, get_attr_kernel->type_vector.size()); in TEST_F()
951 EXPECT_EQ(DT_INT32, get_attr_kernel->type_vector[0]); in TEST_F()
952 EXPECT_EQ(DT_BOOL, get_attr_kernel->type_vector[1]); in TEST_F()
/external/tensorflow/tensorflow/c/eager/
Dc_api.cc1098 absl::InlinedVector<unsigned int, 4> type_vector; in SetOpAttrValueScalar() local
1100 type_vector.push_back(default_value.list().type(i)); in SetOpAttrValueScalar()
1104 reinterpret_cast<const TF_DataType*>(type_vector.data()), in SetOpAttrValueScalar()