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()201 ConstMatrixVector inputs_flat; in Compute() local
109 ConstMatrixVector inputs_flat; in Compute() local
143 ConstMatrixVector inputs_flat; in Compute() local451 ConstMatrixVector inputs_flat; in Compute() local626 ConstMatrixVector inputs_flat; in Compute() local
118 const std::vector<std::unique_ptr<typename TTypes<T, 2>::ConstMatrix>>& in ConcatGPUSlice()
108 ConstMatrixVector inputs_flat; in Compute() local
95 ConstMatrixVector inputs_flat; in Compute() local
53 std::vector<std::unique_ptr<typename TTypes<T, 2>::ConstMatrix>> inputs_flat; in Concat() local