Home
last modified time | relevance | path

Searched refs:ConstMatrix (Results 1 – 25 of 81) sorted by relevance

1234

/external/tensorflow/tensorflow/core/kernels/
Dsparse_utils_test.cc43 TTypes<int32>::ConstMatrix indices_mat(data, 8, 2); in TEST()
54 TTypes<int32>::ConstMatrix indices_mat(data, 15, 2); in TEST()
65 TTypes<int64>::ConstMatrix indices_mat(data, 1, 2); in TEST()
74 TTypes<uint32>::ConstMatrix indices_mat(data, 2, 2); in TEST()
83 TTypes<uint16>::ConstMatrix indices_mat(data, 4, 2); in TEST()
93 TTypes<uint64>::ConstMatrix indices_mat(data, 4, 2); in TEST()
145 TTypes<int32>::ConstMatrix indices_mat(data, 8, 2); in TEST()
154 TTypes<int64>::ConstMatrix indices_mat(data, 8, 2); in TEST()
163 TTypes<int32>::ConstMatrix indices_mat(data, 6, 2); in TEST()
172 TTypes<uint16>::ConstMatrix indices_mat(data, 6, 2); in TEST()
[all …]
Dconcat_lib.h47 const std::vector<std::unique_ptr<typename TTypes<T, 2>::ConstMatrix>>&
54 const std::vector<std::unique_ptr<typename TTypes<T, 2>::ConstMatrix>>&
62 const std::vector<std::unique_ptr<typename TTypes<T, 2>::ConstMatrix>>& \
82 const std::vector<std::unique_ptr<typename TTypes<T, 2>::ConstMatrix>>&
Dconcat_lib_cpu.cc53 const std::vector<std::unique_ptr<typename TTypes<T, 2>::ConstMatrix>>& in ConcatCPU()
68 const std::vector<std::unique_ptr<typename TTypes<T, 2>::ConstMatrix>>&, \
91 const std::vector<std::unique_ptr<typename TTypes<T, 2>::ConstMatrix>>& in ConcatSYCL()
100 const std::vector<std::unique_ptr<typename TTypes<T, 2>::ConstMatrix>>&, \
Dtensor_flag_utils.cc77 const typename TTypes<MatrixType>::ConstMatrix& config_mat, in FindConfigValueForKey()
134 const typename TTypes<MatrixType>::ConstMatrix& config_mat, const K key) { in FindConfigValueForKey()
164 const TTypes<float>::ConstMatrix& config_mat, \
167 const TTypes<float>::ConstMatrix& config_mat, const TypeIndex key); \
169 const TTypes<int64>::ConstMatrix& config_mat, const TypeIndex key);
Dxent_op.h41 typename TTypes<T>::ConstMatrix logits,
42 typename TTypes<T>::ConstMatrix labels,
57 typename TTypes<T>::ConstMatrix logits, in Compute()
58 typename TTypes<T>::ConstMatrix labels, in Compute()
Dsparse_utils.cc28 const typename TTypes<Tindices>::ConstMatrix& indices_mat) { in FindNextDenseRowStartIndex()
76 const typename TTypes<Tindices>::ConstMatrix& indices_mat, in GetStartIndicesOfEachDenseRow()
143 const TTypes<TypeIndex>::ConstMatrix& indices_mat); \
145 const TTypes<TypeIndex>::ConstMatrix& indices_mat, \
Done_hot_op.h34 OneGenerator(const typename TTypes<TI>::ConstMatrix& indices, in OneGenerator()
47 const typename TTypes<TI>::ConstMatrix indices_;
59 const Device& d, const typename TTypes<TI>::ConstMatrix& indices, in Compute()
Dconcat_lib_gpu.h35 const std::vector<std::unique_ptr<typename TTypes<T, 2>::ConstMatrix>>&
50 const std::vector<std::unique_ptr<typename TTypes<T, 2>::ConstMatrix>>& \
55 const std::vector<std::unique_ptr<typename TTypes<T, 2>::ConstMatrix>>& \
Dgather_nd_op_cpu_impl.h45 const Index slice_size, typename TTypes<Index>::ConstMatrix Tindices, in GatherNdSliceGenerator()
86 const typename TTypes<Index>::ConstMatrix Tindices_;
101 typename TTypes<Index>::ConstMatrix Tindices,
147 typename TTypes<Index>::ConstMatrix Tindices, \
Dcompare_and_bitpack_op.cc89 static EIGEN_STRONG_INLINE void Compute(typename TTypes<T>::ConstMatrix input, in Compute()
110 typename TTypes<bool>::ConstMatrix input, in Compute()
144 void operator()(OpKernelContext* c, typename TTypes<T>::ConstMatrix input, in operator ()()
184 OpKernelContext* c, typename TTypes<T>::ConstMatrix input, \
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()
97 const std::vector<std::unique_ptr<typename TTypes<T, 2>::ConstMatrix>>& \
Dsparse_utils.h40 const typename TTypes<Tindices>::ConstMatrix& indices_mat);
50 const typename TTypes<Tindices>::ConstMatrix& indices_mat,
Dtensor_flag_utils.h54 const typename TTypes<MatrixType>::ConstMatrix& config_mat,
63 const typename TTypes<MatrixType>::ConstMatrix& config_mat, const K key);
Dxent_op_gpu.cu.cc38 typename TTypes<T>::ConstMatrix logits, in operator ()()
39 typename TTypes<T>::ConstMatrix labels, in operator ()()
Dsparse_xent_op.h40 return To32Bit(typename TTypes<T>::ConstMatrix(in.data(), in.dimensions())); in To32BitConst()
139 typename TTypes<T>::ConstMatrix logits, in Compute()
162 void operator()(OpKernelContext* ctx, typename TTypes<T>::ConstMatrix logits,
174 typename TTypes<T>::ConstMatrix logits, in Compute()
Dmirror_pad_op.cc85 typename TTypes<Tpaddings>::ConstMatrix paddings = in1.matrix<Tpaddings>(); in Compute()
160 TTypes<Tpaddings>::ConstMatrix, int); \
209 TTypes<Tpaddings>::ConstMatrix, int); \
296 typename TTypes<Tpaddings>::ConstMatrix paddings = in1.matrix<Tpaddings>(); in Compute()
369 TTypes<Tpaddings>::ConstMatrix, int, \
415 TTypes<Tpaddings>::ConstMatrix, int, \
Dsparse_xent_op_gpu.cu.cc42 typename TTypes<T>::ConstMatrix logits, in Compute()
62 void operator()(OpKernelContext* ctx, typename TTypes<T>::ConstMatrix logits, in operator ()()
Dtensor_flag_utils_test.cc238 TTypes<float>::ConstMatrix config_mat(data, 3, 3); in TEST()
256 TTypes<float>::ConstMatrix config_mat(data, 1, 3); in TEST()
274 TTypes<float>::ConstMatrix config_mat(data, 2, 3); in TEST()
292 TTypes<float>::ConstMatrix config_mat(data, 3, 2); in TEST()
/external/tensorflow/tensorflow/contrib/rnn/kernels/
Dlstm_ops.h158 bool use_peephole, typename TTypes<T>::ConstMatrix x,
159 typename TTypes<T>::ConstMatrix cs_prev,
160 typename TTypes<T>::ConstMatrix h_prev,
161 typename TTypes<T>::ConstMatrix w,
183 typename TTypes<T>::ConstMatrix x,
184 typename TTypes<T>::ConstMatrix cs_prev,
185 typename TTypes<T>::ConstMatrix h_prev, typename TTypes<T>::ConstMatrix w,
188 typename TTypes<T>::ConstMatrix i, typename TTypes<T>::ConstMatrix cs,
189 typename TTypes<T>::ConstMatrix f, typename TTypes<T>::ConstMatrix o,
190 typename TTypes<T>::ConstMatrix ci, typename TTypes<T>::ConstMatrix co,
[all …]
Dlstm_ops_gpu.cu.cc224 const float cell_clip, bool use_peephole, typename TTypes<T>::ConstMatrix x, in LSTMBlockCellFpropWithCUDA()
225 typename TTypes<T>::ConstMatrix cs_prev, in LSTMBlockCellFpropWithCUDA()
226 typename TTypes<T>::ConstMatrix h_prev, typename TTypes<T>::ConstMatrix w, in LSTMBlockCellFpropWithCUDA()
250 typename TTypes<T>::ConstMatrix const_xh(xh.data(), xh.dimensions()); in LSTMBlockCellFpropWithCUDA()
358 OpKernelContext* ctx, const GPUDevice& d, typename TTypes<T>::ConstMatrix x, in LSTMBlockCellBpropWithCUDA()
359 typename TTypes<T>::ConstMatrix cs_prev, in LSTMBlockCellBpropWithCUDA()
360 typename TTypes<T>::ConstMatrix h_prev, typename TTypes<T>::ConstMatrix w, in LSTMBlockCellBpropWithCUDA()
363 typename TTypes<T>::ConstMatrix i, typename TTypes<T>::ConstMatrix cs, in LSTMBlockCellBpropWithCUDA()
364 typename TTypes<T>::ConstMatrix f, typename TTypes<T>::ConstMatrix o, in LSTMBlockCellBpropWithCUDA()
365 typename TTypes<T>::ConstMatrix ci, typename TTypes<T>::ConstMatrix co, in LSTMBlockCellBpropWithCUDA()
[all …]
Dgru_ops.h75 OpKernelContext* ctx, const Device& d, typename TTypes<T>::ConstMatrix x, in operator()
76 typename TTypes<T>::ConstMatrix h_prev, in operator()
77 typename TTypes<T>::ConstMatrix w_ru, typename TTypes<T>::ConstMatrix w_c, in operator()
88 typename TTypes<T>::ConstMatrix const_x_h_prev(x_h_prev.data(), in operator()
109 typename TTypes<T>::ConstMatrix const_x_h_prevr(x_h_prevr.data(), in operator()
131 OpKernelContext* ctx, const Device& d, typename TTypes<T>::ConstMatrix x, in operator()
132 typename TTypes<T>::ConstMatrix h_prev, in operator()
133 typename TTypes<T>::ConstMatrix w_ru, typename TTypes<T>::ConstMatrix w_c, in operator()
135 typename TTypes<T>::ConstMatrix r, typename TTypes<T>::ConstMatrix u, in operator()
136 typename TTypes<T>::ConstMatrix c, typename TTypes<T>::ConstMatrix d_h, in operator()
[all …]
Dlstm_ops.cc48 typename TTypes<T>::ConstMatrix x, typename TTypes<T>::ConstMatrix cs_prev, in LSTMBlockCellFpropWithEigen()
49 typename TTypes<T>::ConstMatrix h_prev, typename TTypes<T>::ConstMatrix w, in LSTMBlockCellFpropWithEigen()
62 typename TTypes<T>::ConstMatrix const_xh(xh.data(), xh.dimensions()); in LSTMBlockCellFpropWithEigen()
128 bool use_peephole, typename TTypes<T>::ConstMatrix x, in LSTMBlockCellBpropWithEigen()
129 typename TTypes<T>::ConstMatrix cs_prev, in LSTMBlockCellBpropWithEigen()
130 typename TTypes<T>::ConstMatrix h_prev, typename TTypes<T>::ConstMatrix w, in LSTMBlockCellBpropWithEigen()
133 typename TTypes<T>::ConstMatrix i, typename TTypes<T>::ConstMatrix cs, in LSTMBlockCellBpropWithEigen()
134 typename TTypes<T>::ConstMatrix f, typename TTypes<T>::ConstMatrix o, in LSTMBlockCellBpropWithEigen()
135 typename TTypes<T>::ConstMatrix ci, typename TTypes<T>::ConstMatrix co, in LSTMBlockCellBpropWithEigen()
136 typename TTypes<T>::ConstMatrix cs_grad, in LSTMBlockCellBpropWithEigen()
[all …]
Dgru_ops.cc392 typename TTypes<T>::ConstMatrix x, \
393 typename TTypes<T>::ConstMatrix h_prev, \
394 typename TTypes<T>::ConstMatrix w_ru, \
395 typename TTypes<T>::ConstMatrix w_c, typename TTypes<T>::ConstVec b_ru, \
422 typename TTypes<T>::ConstMatrix x, typename TTypes<T>::ConstMatrix h, \
423 typename TTypes<T>::ConstMatrix w_ru, \
424 typename TTypes<T>::ConstMatrix w_c, typename TTypes<T>::ConstVec b_ru, \
425 typename TTypes<T>::ConstVec b_c, typename TTypes<T>::ConstMatrix r, \
426 typename TTypes<T>::ConstMatrix u, typename TTypes<T>::ConstMatrix c, \
427 typename TTypes<T>::ConstMatrix d_h, typename TTypes<T>::Matrix d_x, \
Dblas_gemm.h56 typename TTypes<T>::ConstMatrix a,
57 typename TTypes<T>::ConstMatrix b,
74 typename TTypes<T>::ConstMatrix a,
75 typename TTypes<T>::ConstMatrix b,
/external/tensorflow/tensorflow/contrib/boosted_trees/lib/utils/
Dsparse_column_iterable.h102 SparseColumnIterable(TTypes<int64>::ConstMatrix ix, int64 example_start, in SparseColumnIterable()
111 const TTypes<int64>::ConstMatrix& ix() const { return ix_; } in ix()
115 const TTypes<int64>::ConstMatrix& sparse_indices() const { return ix_; } in sparse_indices()
119 TTypes<int64>::ConstMatrix ix_;

1234