Home
last modified time | relevance | path

Searched refs:ConstMatrix (Results 1 – 25 of 92) 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_cpu.cc50 const std::vector<std::unique_ptr<typename TTypes<T, 2>::ConstMatrix>>& in EstimateBytesPerElement()
62 std::unique_ptr<typename TTypes<tstring, 2>::ConstMatrix>>& inputs) { in EstimateBytesPerElement()
101 const std::vector<std::unique_ptr<typename TTypes<T, 2>::ConstMatrix>>& in ConcatCPU()
111 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);
Dconcat_lib.h47 const std::vector<std::unique_ptr<typename TTypes<T, 2>::ConstMatrix>>&
55 const std::vector<std::unique_ptr<typename TTypes<T, 2>::ConstMatrix>>&
63 const std::vector<std::unique_ptr<typename TTypes<T, 2>::ConstMatrix>>& \
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()
146 const TTypes<TypeIndex>::ConstMatrix& indices_mat); \
148 const TTypes<TypeIndex>::ConstMatrix& indices_mat, \
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>>& \
Done_hot_op.h40 OneGenerator(const typename TTypes<TI>::ConstMatrix& indices, in OneGenerator()
53 const typename TTypes<TI>::ConstMatrix indices_;
65 const Device& d, const typename TTypes<TI>::ConstMatrix& indices, in Compute()
77 const CPUDevice& d, const typename TTypes<TI>::ConstMatrix& indices,
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,
131 typename TTypes<Index>::ConstMatrix Tindices, \
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,
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, \
Dxent_op_gpu.cu.cc39 typename TTypes<T>::ConstMatrix logits, in operator ()()
40 typename TTypes<T>::ConstMatrix labels, in operator ()()
Dtensor_flag_utils.h54 const typename TTypes<MatrixType>::ConstMatrix& config_mat,
63 const typename TTypes<MatrixType>::ConstMatrix& config_mat, const K key);
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()
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 ()()
Dsparse_tensor_dense_matmul_op.h33 typename TTypes<Tindices>::ConstMatrix a_indices,
34 typename TTypes<T>::ConstVec a_values, typename TTypes<T>::ConstMatrix b);
/external/tensorflow/tensorflow/core/kernels/rnn/
Dlstm_ops.h170 bool use_peephole, typename TTypes<T>::ConstMatrix x,
171 typename TTypes<T>::ConstMatrix cs_prev,
172 typename TTypes<T>::ConstMatrix h_prev,
173 typename TTypes<T>::ConstMatrix w,
195 typename TTypes<T>::ConstMatrix x,
196 typename TTypes<T>::ConstMatrix cs_prev,
197 typename TTypes<T>::ConstMatrix h_prev, typename TTypes<T>::ConstMatrix w,
200 typename TTypes<T>::ConstMatrix i, typename TTypes<T>::ConstMatrix cs,
201 typename TTypes<T>::ConstMatrix f, typename TTypes<T>::ConstMatrix o,
202 typename TTypes<T>::ConstMatrix ci, typename TTypes<T>::ConstMatrix co,
[all …]
Dlstm_ops_gpu.cu.cc229 const float cell_clip, bool use_peephole, typename TTypes<T>::ConstMatrix x, in LSTMBlockCellFpropWithCUDA()
230 typename TTypes<T>::ConstMatrix cs_prev, in LSTMBlockCellFpropWithCUDA()
231 typename TTypes<T>::ConstMatrix h_prev, typename TTypes<T>::ConstMatrix w, in LSTMBlockCellFpropWithCUDA()
255 typename TTypes<T>::ConstMatrix const_xh(xh.data(), xh.dimensions()); in LSTMBlockCellFpropWithCUDA()
365 OpKernelContext* ctx, const GPUDevice& d, typename TTypes<T>::ConstMatrix x, in LSTMBlockCellBpropWithCUDA()
366 typename TTypes<T>::ConstMatrix cs_prev, in LSTMBlockCellBpropWithCUDA()
367 typename TTypes<T>::ConstMatrix h_prev, typename TTypes<T>::ConstMatrix w, in LSTMBlockCellBpropWithCUDA()
370 typename TTypes<T>::ConstMatrix i, typename TTypes<T>::ConstMatrix cs, in LSTMBlockCellBpropWithCUDA()
371 typename TTypes<T>::ConstMatrix f, typename TTypes<T>::ConstMatrix o, in LSTMBlockCellBpropWithCUDA()
372 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()
132 bool use_peephole, typename TTypes<T>::ConstMatrix x, in LSTMBlockCellBpropWithEigen()
133 typename TTypes<T>::ConstMatrix cs_prev, in LSTMBlockCellBpropWithEigen()
134 typename TTypes<T>::ConstMatrix h_prev, typename TTypes<T>::ConstMatrix w, in LSTMBlockCellBpropWithEigen()
137 typename TTypes<T>::ConstMatrix i, typename TTypes<T>::ConstMatrix cs, in LSTMBlockCellBpropWithEigen()
138 typename TTypes<T>::ConstMatrix f, typename TTypes<T>::ConstMatrix o, in LSTMBlockCellBpropWithEigen()
139 typename TTypes<T>::ConstMatrix ci, typename TTypes<T>::ConstMatrix co, in LSTMBlockCellBpropWithEigen()
140 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/core/kernels/sparse/
Dkernels.h39 Status operator()(OpKernelContext* c, TTypes<int64>::ConstMatrix indices,
60 TTypes<int64>::ConstMatrix indices,
123 TTypes<int64>::ConstMatrix indices,
149 typename TTypes<T>::ConstMatrix b,
/external/tensorflow/tensorflow/core/kernels/boosted_trees/
Dprediction_ops.cc41 std::vector<tensorflow::TTypes<int32>::ConstMatrix>& bucketized_features) { in ConvertVectorsToMatrices()
46 TTypes<int32>::ConstMatrix(v.data(), v.size(), 1)); in ConvertVectorsToMatrices()
77 std::vector<tensorflow::TTypes<int32>::ConstMatrix> bucketized_features; in Compute()
220 std::vector<tensorflow::TTypes<int32>::ConstMatrix> bucketized_features; in Compute()
320 std::vector<tensorflow::TTypes<int32>::ConstMatrix> bucketized_features; in Compute()

1234