Searched refs:DefineTensor (Results 1 – 6 of 6) sorted by relevance
148 #define DefineTensor(X, T) std::vector<T> X##_; macro150 FOR_ALL_INPUT_AND_WEIGHT_TENSORS(DefineTensor);151 FOR_ALL_OUTPUT_TENSORS(DefineTensor);153 #undef DefineTensor
161 #define DefineTensor(X, T) std::vector<T> X##_; macro163 FOR_ALL_INPUT_AND_WEIGHT_TENSORS(DefineTensor);164 FOR_ALL_OUTPUT_TENSORS(DefineTensor);166 #undef DefineTensor
231 #define DefineTensor(X) std::vector<float> X##_; macro233 FOR_ALL_INPUT_AND_WEIGHT_TENSORS(DefineTensor);234 FOR_ALL_OUTPUT_TENSORS(DefineTensor);236 #undef DefineTensor
277 #define DefineTensor(X) std::vector<float> X##_; macro279 FOR_ALL_INPUT_AND_WEIGHT_TENSORS(DefineTensor);280 FOR_ALL_LAYER_NORM_WEIGHTS(DefineTensor);281 FOR_ALL_OUTPUT_TENSORS(DefineTensor);283 #undef DefineTensor
308 #define DefineTensor(X) std::vector<float> X##_; macro310 FOR_ALL_INPUT_AND_WEIGHT_TENSORS(DefineTensor);311 FOR_ALL_OUTPUT_TENSORS(DefineTensor);313 #undef DefineTensor
264 #define DefineTensor(X) std::vector<float> X##_; macro266 FOR_ALL_INPUT_AND_WEIGHT_TENSORS(DefineTensor);267 FOR_ALL_OUTPUT_TENSORS(DefineTensor);269 #undef DefineTensor