Home
last modified time | relevance | path

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

12

/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()
76 typename TTypes<T>::ConstVec mean, in operator()
77 typename TTypes<T>::ConstVec var, in operator()
78 typename TTypes<T>::ConstVec gamma, in operator()
88 typedef typename TTypes<T>::ConstVec::Index Index; in operator()
Dquantize_and_dequantize_op.h45 void operator()(const Device& d, typename TTypes<T>::ConstVec input,
53 void ClampScaleAndRound(const Device& d, typename TTypes<T>::ConstVec input, in ClampScaleAndRound()
63 void ClampScaleAndRound(const Device& d, typename TTypes<T>::ConstVec input, in ClampScaleAndRound()
81 void ScaleAndRound(const Device& d, typename TTypes<T>::ConstVec input, T scale, in ScaleAndRound()
89 void ScaleAndRound(const Device& d, typename TTypes<T>::ConstVec input, T scale, in ScaleAndRound()
107 static void Compute(const Device& d, typename TTypes<T>::ConstVec input, in Compute()
Dbatch_norm_op.cc185 typename TTypes<T>::ConstVec mean, typename TTypes<T>::ConstVec var, \
186 typename TTypes<T>::ConstVec beta, typename TTypes<T>::ConstVec gamma, \
241 typename TTypes<T>::ConstVec mean, typename TTypes<T>::ConstVec var, \
242 typename TTypes<T>::ConstVec gamma, \
Dreverse_sequence_op.h33 int32 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()
Dfake_quant_ops_functor.h68 using ConstVec = typename tensorflow::TTypes<T>::ConstVec; variable
201 ConstVec<float> min, ConstVec<float> max, const int quant_min, in operator()
234 TTypes<float>::ConstMatrix inputs, ConstVec<float> min, in operator()
235 ConstVec<float> max, const int quant_min, const int quant_max, in operator()
Dfused_batch_norm_op.h75 typename TTypes<U>::ConstVec scale(scale_input.vec<U>()); in operator()
76 typename TTypes<U>::ConstVec pop_mean(pop_mean_input.vec<U>()); in operator()
77 typename TTypes<U>::ConstVec pop_var(pop_variance_input.vec<U>()); in operator()
Dsparse_xent_op.h34 return To32Bit(typename TTypes<T>::ConstVec(in.data(), in.dimensions())); in To32BitConst()
163 typename TTypes<Index>::ConstVec labels,
175 typename TTypes<Index>::ConstVec labels, in Compute()
Dfused_batch_norm_op.cc79 typename TTypes<U>::ConstVec scale(scale_input.vec<U>()); in operator ()()
80 typename TTypes<U>::ConstVec offset(offset_input.vec<U>()); in operator ()()
81 typename TTypes<U>::ConstVec estimated_mean(estimated_mean_input.vec<U>()); in operator ()()
82 typename TTypes<U>::ConstVec estimated_variance( in operator ()()
167 typename TTypes<U>::ConstVec scale(scale_input.vec<U>()); in operator ()()
168 typename TTypes<U>::ConstVec mean(mean_input.vec<U>()); in operator ()()
169 typename TTypes<U>::ConstVec variance(variance_input.vec<U>()); in operator ()()
Dquantize_and_dequantize_op_gpu.cu.cc32 void operator()(const GPUDevice& d, typename TTypes<T>::ConstVec input, in operator ()()
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 ()()
Dsparse_tensor_dense_matmul_op.h34 typename TTypes<T>::ConstVec a_values, typename TTypes<T>::ConstMatrix b);
Dsparse_tensor_dense_matmul_op.cc186 typename TTypes<T>::ConstVec a_values, \
249 typename TTypes<T>::ConstVec a_values, in Compute()
Dcwise_op_gpu_select.cu.cc61 TTypes<bool>::ConstVec cond_vec, in operator ()()
/external/tensorflow/tensorflow/contrib/rnn/kernels/
Dlstm_ops_gpu.cu.cc227 typename TTypes<T>::ConstVec wci, typename TTypes<T>::ConstVec wcf, in LSTMBlockCellFpropWithCUDA()
228 typename TTypes<T>::ConstVec wco, typename TTypes<T>::ConstVec b, in LSTMBlockCellFpropWithCUDA()
361 typename TTypes<T>::ConstVec wci, typename TTypes<T>::ConstVec wcf, in LSTMBlockCellBpropWithCUDA()
362 typename TTypes<T>::ConstVec wco, typename TTypes<T>::ConstVec b, in LSTMBlockCellBpropWithCUDA()
416 typename TTypes<T>::ConstMatrix w, typename TTypes<T>::ConstVec wci, \
417 typename TTypes<T>::ConstVec wcf, typename TTypes<T>::ConstVec wco, \
418 typename TTypes<T>::ConstVec b, typename TTypes<T>::Matrix xh, \
434 typename TTypes<T>::ConstMatrix w, typename TTypes<T>::ConstVec wci, \
435 typename TTypes<T>::ConstVec wcf, typename TTypes<T>::ConstVec wco, \
436 typename TTypes<T>::ConstVec b, typename TTypes<T>::ConstMatrix i, \
Dlstm_ops.h79 typename TTypes<int64>::ConstVec seq_len, in operator()
162 typename TTypes<T>::ConstVec wci,
163 typename TTypes<T>::ConstVec wcf,
164 typename TTypes<T>::ConstVec wco,
165 typename TTypes<T>::ConstVec b, typename TTypes<T>::Matrix xh,
186 typename TTypes<T>::ConstVec wci, typename TTypes<T>::ConstVec wcf,
187 typename TTypes<T>::ConstVec wco, typename TTypes<T>::ConstVec b,
211 typename TTypes<T>::ConstVec wci, typename TTypes<T>::ConstVec wcf, in operator()
212 typename TTypes<T>::ConstVec wco, typename TTypes<T>::ConstVec b, in operator()
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()
131 typename TTypes<T>::ConstVec wci, typename TTypes<T>::ConstVec wcf, in LSTMBlockCellBpropWithEigen()
132 typename TTypes<T>::ConstVec wco, typename TTypes<T>::ConstVec b, in LSTMBlockCellBpropWithEigen()
189 typename TTypes<T>::ConstMatrix w, typename TTypes<T>::ConstVec wci, \
190 typename TTypes<T>::ConstVec wcf, typename TTypes<T>::ConstVec wco, \
191 typename TTypes<T>::ConstVec b, typename TTypes<T>::Matrix xh, \
206 typename TTypes<T>::ConstMatrix w, typename TTypes<T>::ConstVec wci, \
207 typename TTypes<T>::ConstVec wcf, typename TTypes<T>::ConstVec wco, \
208 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/contrib/seq2seq/kernels/
Dbeam_search_ops.cc85 typename TTypes<int32>::ConstVec max_seq_lens_t = in Compute()
110 TTypes<int32>::ConstVec max_sequence_lengths, in operator ()()
176 TTypes<int32>::ConstVec max_sequence_lengths, const T end_token, \
Dbeam_search_ops.h33 TTypes<int32>::ConstVec max_sequence_lengths,
/external/tensorflow/tensorflow/contrib/boosted_trees/lib/utils/
Dexamples_iterable.h192 std::vector<TTypes<float>::ConstVec> sparse_float_column_values_;
201 std::vector<TTypes<int64>::ConstVec> sparse_int_column_values_;
/external/tensorflow/tensorflow/core/kernels/boosted_trees/
Dprediction_ops.cc68 std::vector<tensorflow::TTypes<int32>::ConstVec> batch_bucketized_features; in Compute()
207 std::vector<tensorflow::TTypes<int32>::ConstVec> batch_bucketized_features; in Compute()
311 std::vector<tensorflow::TTypes<int32>::ConstVec> batch_bucketized_features; in Compute()
/external/tensorflow/tensorflow/contrib/boosted_trees/kernels/
Dsplit_handler_ops.cc245 const tensorflow::TTypes<float>::ConstVec& bucket_boundaries, in ComputeNormalDecisionTree()
246 const tensorflow::TTypes<int32>::ConstVec& partition_ids, in ComputeNormalDecisionTree()
305 const tensorflow::TTypes<float>::ConstVec& bucket_boundaries, in ComputeObliviousDecisionTree()
306 const tensorflow::TTypes<int32>::ConstVec& partition_ids, in ComputeObliviousDecisionTree()
816 const tensorflow::TTypes<int32>::ConstVec& partition_ids, in ComputeNormalDecisionTree()
889 const tensorflow::TTypes<int32>::ConstVec& partition_ids, in ComputeObliviousDecisionTree()

12