Home
last modified time | relevance | path

Searched refs:ConstVec (Results 1 – 25 of 62) sorted by relevance

123

/external/tensorflow/tensorflow/core/kernels/
Dbatch_norm_op.h29 typename TTypes<T>::ConstVec mean, in operator()
30 typename TTypes<T>::ConstVec var, in operator()
31 typename TTypes<T>::ConstVec beta, in operator()
32 typename TTypes<T>::ConstVec gamma, T variance_epsilon, in operator()
70 typename TTypes<T>::ConstVec mean, in operator()
71 typename TTypes<T>::ConstVec var, in operator()
72 typename TTypes<T>::ConstVec gamma, in operator()
82 typedef typename TTypes<T>::ConstVec::Index Index; in operator()
Dquantize_and_dequantize_op.h47 void operator()(const Device& d, typename TTypes<T>::ConstVec input,
88 typename ConstVec = typename TTypes<T>::ConstVec>
89 void ClampScaleAndRound(const Device& d, ConstVec input, T min_range, in ClampScaleAndRound()
99 typename ConstVec = typename TTypes<T>::ConstVec>
100 void ClampScaleAndRound(const Device& d, ConstVec input, T min_range, in ClampScaleAndRound()
119 typename ConstVec = typename TTypes<T>::ConstVec>
120 void ScaleAndRound(const Device& d, ConstVec input, T scale, T inverse_scale, in ScaleAndRound()
127 typename ConstVec = typename TTypes<T>::ConstVec>
128 void ScaleAndRound(const Device& d, ConstVec input, T scale, T inverse_scale, in ScaleAndRound()
182 static void Compute(const Device& d, typename TTypes<T>::ConstVec input, in Compute()
Dfused_batch_norm_op.h46 typename TTypes<U>::ConstVec scale,
47 typename TTypes<U>::ConstVec offset,
48 typename TTypes<U>::ConstVec estimated_mean,
49 typename TTypes<U>::ConstVec estimated_variance,
Dbatch_norm_op.cc182 typename TTypes<T>::ConstVec mean, typename TTypes<T>::ConstVec var, \
183 typename TTypes<T>::ConstVec beta, typename TTypes<T>::ConstVec gamma, \
227 typename TTypes<T>::ConstVec mean, typename TTypes<T>::ConstVec var, \
228 typename TTypes<T>::ConstVec gamma, \
Dreverse_sequence_op.h33 int32_t seq_dim, typename TTypes<Tlen>::ConstVec seq_lengths) in ReverseGenerator()
54 typename TTypes<Tlen>::ConstVec seq_lengths_;
66 typename TTypes<Tlen>::ConstVec seq_lengths, in Compute()
Dsegment_reduction_ops.h142 typename TTypes<Index>::ConstVec indices,
143 typename TTypes<SegmentId>::ConstVec segment_ids,
152 typename TTypes<Index>::ConstVec indices_vec,
153 typename TTypes<SegmentId>::ConstVec segment_vec,
Dfake_quant_ops_functor.h71 using ConstVec = typename tensorflow::TTypes<T>::ConstVec; variable
208 ConstVec<float> min, ConstVec<float> max, const int quant_min, in operator()
244 TTypes<float>::ConstMatrix inputs, ConstVec<float> min, in operator()
245 ConstVec<float> max, const int quant_min, const int quant_max, in operator()
Dsparse_fill_empty_rows_op.h39 typename TTypes<Tindex>::ConstVec reverse_index_map,
40 typename TTypes<T>::ConstVec grad_values,
Dfused_batch_norm_op.cu.cc46 typename TTypes<U>::ConstVec scale(scale_input.vec<U>()); in operator ()()
47 typename TTypes<U>::ConstVec pop_mean(pop_mean_input.vec<U>()); in operator ()()
48 typename TTypes<U>::ConstVec pop_var(pop_variance_input.vec<U>()); in operator ()()
319 typename TTypes<U>::ConstVec scale, in operator ()()
320 typename TTypes<U>::ConstVec offset, in operator ()()
321 typename TTypes<U>::ConstVec estimated_mean, in operator ()()
322 typename TTypes<U>::ConstVec estimated_variance, in operator ()()
Din_topk_op.h49 typename TTypes<TargetT>::ConstVec targets, const TopKArg k, in operator()
57 typename TTypes<TargetT>::ConstVec targets, const TopKArg k,
Dfused_batch_norm_op.cc155 typename TTypes<U>::ConstVec scale(scale_input.vec<U>()); in operator ()()
156 typename TTypes<U>::ConstVec offset(offset_input.vec<U>()); in operator ()()
157 typename TTypes<U>::ConstVec old_mean(running_mean_input.vec<U>()); in operator ()()
158 typename TTypes<U>::ConstVec old_variance(running_variance_input.vec<U>()); in operator ()()
296 typename TTypes<U>::ConstVec scale(scale_input.vec<U>()); in operator ()()
297 typename TTypes<U>::ConstVec offset(offset_input.vec<U>()); in operator ()()
298 typename TTypes<U>::ConstVec estimated_mean(estimated_mean_input.vec<U>()); in operator ()()
299 typename TTypes<U>::ConstVec estimated_variance( in operator ()()
409 typename TTypes<U>::ConstVec scale(scale_input.vec<U>()); in operator ()()
410 typename TTypes<U>::ConstVec mean(mean_input.vec<U>()); in operator ()()
[all …]
Dsparse_xent_op.h34 return To32Bit(typename TTypes<T>::ConstVec(in.data(), in.dimensions())); in To32BitConst()
158 typename TTypes<Index>::ConstVec labels,
170 typename TTypes<Index>::ConstVec labels, in Compute()
Dsparse_fill_empty_rows_op.cc336 typename TTypes<Tindex>::ConstVec reverse_index_map, in operator ()()
337 typename TTypes<T>::ConstVec grad_values, in operator ()()
441 typename TTypes<Tindex>::ConstVec reverse_index_map, \
442 typename TTypes<T>::ConstVec grad_values, \
Dsparse_tensor_dense_matmul_op.cc188 typename TTypes<T>::ConstVec a_values, \
257 typename TTypes<T>::ConstVec a_values, typename TTypes<T>::ConstMatrix b) { in SparseTensorDenseMatMulImpl()
330 typename TTypes<T>::ConstVec a_values, in Compute()
Dbias_op.h31 typename TTypes<T>::ConstVec bias, in operator()
Dsummary_image_op.cc122 typename TTypes<uint8>::ConstVec bad_color(bad_color_full.data(), depth); in NormalizeAndAddImages()
182 typename TTypes<uint8>::ConstVec bad_color, in NormalizeFloatImage()
Dsparse_xent_op_gpu.cu.cc63 typename TTypes<Index>::ConstVec labels, in operator ()()
/external/tensorflow/tensorflow/core/kernels/rnn/
Dlstm_ops.h89 typename TTypes<int64_t>::ConstVec seq_len, in operator()
174 typename TTypes<T>::ConstVec wci,
175 typename TTypes<T>::ConstVec wcf,
176 typename TTypes<T>::ConstVec wco,
177 typename TTypes<T>::ConstVec b, typename TTypes<T>::Matrix xh,
198 typename TTypes<T>::ConstVec wci, typename TTypes<T>::ConstVec wcf,
199 typename TTypes<T>::ConstVec wco, typename TTypes<T>::ConstVec b,
223 typename TTypes<T>::ConstVec wci, typename TTypes<T>::ConstVec wcf, in operator()
224 typename TTypes<T>::ConstVec wco, typename TTypes<T>::ConstVec b, in operator()
Dlstm_ops_gpu.cu.cc232 typename TTypes<T>::ConstVec wci, typename TTypes<T>::ConstVec wcf, in LSTMBlockCellFpropWithCUDA()
233 typename TTypes<T>::ConstVec wco, typename TTypes<T>::ConstVec b, in LSTMBlockCellFpropWithCUDA()
368 typename TTypes<T>::ConstVec wci, typename TTypes<T>::ConstVec wcf, in LSTMBlockCellBpropWithCUDA()
369 typename TTypes<T>::ConstVec wco, typename TTypes<T>::ConstVec b, in LSTMBlockCellBpropWithCUDA()
418 typename TTypes<T>::ConstMatrix w, typename TTypes<T>::ConstVec wci, \
419 typename TTypes<T>::ConstVec wcf, typename TTypes<T>::ConstVec wco, \
420 typename TTypes<T>::ConstVec b, typename TTypes<T>::Matrix xh, \
437 typename TTypes<T>::ConstMatrix w, typename TTypes<T>::ConstVec wci, \
438 typename TTypes<T>::ConstVec wcf, typename TTypes<T>::ConstVec wco, \
439 typename TTypes<T>::ConstVec b, typename TTypes<T>::ConstMatrix i, \
Dlstm_ops.cc50 typename TTypes<T>::ConstVec wci, typename TTypes<T>::ConstVec wcf, in LSTMBlockCellFpropWithEigen()
51 typename TTypes<T>::ConstVec wco, typename TTypes<T>::ConstVec b, in LSTMBlockCellFpropWithEigen()
135 typename TTypes<T>::ConstVec wci, typename TTypes<T>::ConstVec wcf, in LSTMBlockCellBpropWithEigen()
136 typename TTypes<T>::ConstVec wco, typename TTypes<T>::ConstVec b, in LSTMBlockCellBpropWithEigen()
196 typename TTypes<T>::ConstMatrix w, typename TTypes<T>::ConstVec wci, \
197 typename TTypes<T>::ConstVec wcf, typename TTypes<T>::ConstVec wco, \
198 typename TTypes<T>::ConstVec b, typename TTypes<T>::Matrix xh, \
214 typename TTypes<T>::ConstMatrix w, typename TTypes<T>::ConstVec wci, \
215 typename TTypes<T>::ConstVec wcf, typename TTypes<T>::ConstVec wco, \
216 typename TTypes<T>::ConstVec b, typename TTypes<T>::ConstMatrix i, \
[all …]
Dgru_ops.h78 typename TTypes<T>::ConstVec b_ru, typename TTypes<T>::ConstVec b_c, in operator()
134 typename TTypes<T>::ConstVec b_ru, typename TTypes<T>::ConstVec b_c, in operator()
Dgru_ops.cc395 typename TTypes<T>::ConstMatrix w_c, typename TTypes<T>::ConstVec b_ru, \
396 typename TTypes<T>::ConstVec b_c, typename TTypes<T>::Matrix r_u_bar, \
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, \
/external/tensorflow/tensorflow/core/kernels/sparse/
Dkernels.h59 void operator()(const Device& d, TTypes<int64_t>::ConstVec host_dense_shape,
86 TTypes<int64_t>::ConstVec host_dense_shape,
87 TTypes<int32>::ConstVec host_batch_ptrs,
89 TTypes<int32>::ConstVec coo_col_ind,
Dcsr_sparse_matrix_to_sparse_tensor_op.cc226 TTypes<int64_t>::ConstVec host_dense_shape,
227 TTypes<int>::ConstVec host_batch_ptrs,
229 TTypes<int>::ConstVec coo_col_ind,
Dcsr_sparse_matrix_to_dense_op.cc241 TTypes<int64_t>::ConstVec host_dense_shape,
242 TTypes<int>::ConstVec host_batch_ptrs,
244 TTypes<int>::ConstVec coo_col_ind,

123