Home
last modified time | relevance | path

Searched defs:vectorizer (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/optimizers/data/vectorization/
Dvectorizer_registry.h49 std::unique_ptr<Vectorizer> vectorizer) { in VectorizerRegistration()
56 #define REGISTER_VECTORIZER(op_type, vectorizer) \ argument
59 #define REGISTER_VECTORIZER_UNIQ_HELPER(ctr, op_type, vectorizer) \ argument
62 #define REGISTER_VECTORIZER_UNIQ(ctr, op_type, vectorizer) \ argument
Dvectorizer_registry.cc37 std::unique_ptr<Vectorizer> vectorizer) { in Register()
Dvectorizer_registry_test.cc37 auto vectorizer = VectorizerRegistry::Global()->Get("test_op"); in TEST() local
/external/tensorflow/tensorflow/core/grappler/optimizers/data/
Dvectorization_utils.cc240 auto vectorizer = VectorizerRegistry::Global()->Get(op_node->type_string()); in AddConversionMapping() local