Home
last modified time | relevance | path

Searched defs:inputs_flat (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dconcat_lib_gpu.cc38 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()
Dquantized_concat_op.cc121 ConstMatrixVector* inputs_flat, in CalculateConcatDims()
199 ConstMatrixVector inputs_flat; in Compute() local
Dpack_op.cc96 ConstMatrixVector inputs_flat; in Compute() local
Dlist_kernels.h122 ConstMatrixVector inputs_flat; in Compute() local
426 ConstMatrixVector inputs_flat; in Compute() local
603 ConstMatrixVector inputs_flat; in Compute() local
Dconcat_lib_gpu_impl.cu.cc118 const std::vector<std::unique_ptr<typename TTypes<T, 2>::ConstMatrix>>& in ConcatGPUSlice()
Dconcat_op.cc115 ConstMatrixVector inputs_flat; in Compute() local
/external/tensorflow/tensorflow/core/kernels/batching_util/
Dconcat_split_util.h46 std::vector<std::unique_ptr<typename TTypes<T, 2>::ConstMatrix>> inputs_flat; in Concat() local
/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_concat_op.cc174 ConstMatrixVector inputs_flat; in Compute() local