Searched defs:inputs_flat (Results 1 – 8 of 8) sorted by relevance
38 const std::vector<std::unique_ptr<typename TTypes<T, 2>::ConstMatrix>>& in ConcatGPUCall()73 const std::vector<std::unique_ptr<typename TTypes<T, 2>::ConstMatrix>>& in ConcatGPU()
121 ConstMatrixVector* inputs_flat, in CalculateConcatDims()200 ConstMatrixVector inputs_flat; in Compute() local
227 ConstMatrixVector inputs_flat; in Compute() local531 ConstMatrixVector inputs_flat; in Compute() local715 ConstMatrixVector inputs_flat; in Compute() local
94 ConstMatrixVector inputs_flat; in Compute() local
118 const std::vector<std::unique_ptr<typename TTypes<T, 2>::ConstMatrix>>& in ConcatGPUSlice()
114 ConstMatrixVector inputs_flat; in Compute() local
46 std::vector<std::unique_ptr<typename TTypes<T, 2>::ConstMatrix>> inputs_flat; in Concat() local
178 ConstMatrixVector inputs_flat; in Compute() local