Home
last modified time | relevance | path

Searched refs:VectorizerInput (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/optimizers/data/vectorization/
Dvectorizer.h35 class VectorizerInput {
37 VectorizerInput(std::vector<WrappedTensor>&& inputs) in VectorizerInput() function
110 VectorizerInput&& inputs,
Dcwise_op_vectorizer.cc46 Status ExpandDimsForBroadcast(VectorizerInput* inputs, Graph* g) { in ExpandDimsForBroadcast()
108 *inputs = VectorizerInput({expand_dims_if_unstacked(inputs->at(0), rank_0), in ExpandDimsForBroadcast()
116 VectorizerInput&& inputs, in CwiseVectorizeHelper()
138 VectorizerInput&& inputs, in Vectorize()
153 VectorizerInput&& inputs, in Vectorize()
Dvectorizer_registry_test.cc26 VectorizerInput&& inputs, in Vectorize()
Dunpack_vectorizer.cc27 VectorizerInput&& inputs, in Vectorize()
Ddecode_csv_vectorizer.cc28 VectorizerInput&& inputs, in Vectorize()
Dtranspose_vectorizer.cc44 VectorizerInput&& inputs, in Vectorize()
Dreshape_vectorizer.cc50 VectorizerInput&& inputs, in Vectorize()
Dparse_single_example_vectorizer.cc30 VectorizerInput&& inputs, in Vectorize()