| /external/tensorflow/tensorflow/core/kernels/ |
| D | training_ops.h | 33 typename TTypes<T>::ConstScalar alpha,
42 typename TTypes<T>::ConstScalar lr,
43 typename TTypes<T>::ConstScalar rho,
44 typename TTypes<T>::ConstScalar epsilon,
51 typename TTypes<T>::ConstScalar lr,
52 typename TTypes<T>::ConstScalar l1,
53 typename TTypes<T>::ConstScalar l2,
60 typename TTypes<T>::ConstScalar lr,
61 typename TTypes<T>::ConstScalar l1,
62 typename TTypes<T>::ConstScalar l2,
[all …]
|
| D | training_ops_gpu.cu.cc | 31 typename TTypes<T>::ConstScalar lr, in operator ()()
44 typename TTypes<T>::ConstScalar lr, in operator ()()
61 typename TTypes<T>::ConstScalar lr, in operator ()()
62 typename TTypes<T>::ConstScalar rho, in operator ()()
63 typename TTypes<T>::ConstScalar epsilon, in operator ()()
87 typename TTypes<T>::ConstScalar lr, in operator ()()
89 typename TTypes<T>::ConstScalar momentum, bool use_nesterov) { in operator ()()
108 typename TTypes<T>::ConstScalar lr, in operator ()()
110 typename TTypes<T>::ConstScalar momentum, bool use_nesterov) { in operator ()()
129 typename TTypes<T>::ConstScalar beta1_power, in operator ()()
[all …]
|
| D | one_hot_op.h | 35 const typename TTypes<T>::ConstScalar& on_value, in OneGenerator()
36 const typename TTypes<T>::ConstScalar& off_value) in OneGenerator()
48 const typename TTypes<T>::ConstScalar on_value_;
49 const typename TTypes<T>::ConstScalar off_value_;
60 const typename TTypes<T>::ConstScalar& on_value, in Compute()
61 const typename TTypes<T>::ConstScalar& off_value, in Compute()
|
| D | training_ops.cc | 51 typename TTypes<T>::ConstScalar lr, in operator ()()
72 typename TTypes<T>::ConstScalar lr, in operator ()()
73 typename TTypes<T>::ConstScalar rho, in operator ()()
74 typename TTypes<T>::ConstScalar epsilon, in operator ()()
89 typename TTypes<T>::ConstScalar lr, in operator ()()
90 typename TTypes<T>::ConstScalar l1, in operator ()()
91 typename TTypes<T>::ConstScalar l2, in operator ()()
118 typename TTypes<T>::ConstScalar lr, int64 global_step, in operator ()()
119 typename TTypes<T>::ConstScalar l1, in operator ()()
120 typename TTypes<T>::ConstScalar l2, in operator ()()
[all …]
|
| D | fake_quant_ops_functor.h | 64 using ConstScalar = typename tensorflow::TTypes<T>::ConstScalar; variable
128 ConstScalar<float> min, ConstScalar<float> max, in operator()
156 ConstFlat<float> inputs, ConstScalar<float> min, in operator()
157 ConstScalar<float> max, const int quant_min, in operator()
|
| D | adjust_contrast_op.h | 28 typename TTypes<float>::ConstScalar contrast_factor, in operator()
29 typename TTypes<float>::ConstScalar min_value, in operator()
30 typename TTypes<float>::ConstScalar max_value, in operator()
93 typename TTypes<float>::ConstScalar contrast_factor, in operator()
|
| D | adjust_contrast_op.cc | 110 typename TTypes<float>::ConstScalar contrast_factor, \
111 typename TTypes<float>::ConstScalar min_value, \
112 typename TTypes<float>::ConstScalar max_value, \
370 typename TTypes<float>::ConstScalar factor, in IncrementWithScaling()
394 typename TTypes<float>::ConstScalar contrast_factor, \
|
| D | scatter_functor.h | 351 const typename TTypes<T>::ConstScalar update,
359 const typename TTypes<T>::ConstScalar update,
382 const typename TTypes<Variant>::ConstScalar update,
418 const typename TTypes<T>::ConstScalar update,
443 const typename TTypes<T>::ConstScalar update,
471 const typename TTypes<T>::ConstScalar update,
|
| D | fill_functor.cc | 128 typename TTypes<T>::ConstScalar in) { in operator ()()
146 typename TTypes<T>::ConstScalar in) { in operator ()()
|
| D | fake_quant_ops.cc | 272 typename TTypes<float>::ConstScalar min,
273 typename TTypes<float>::ConstScalar max, const int quant_min,
286 typename TTypes<float>::ConstScalar min,
287 typename TTypes<float>::ConstScalar max, const int quant_min,
|
| D | one_hot_op.cc | 151 const typename TTypes<T>::ConstScalar& on_value, \
152 const typename TTypes<T>::ConstScalar& off_value, \
|
| D | compare_and_bitpack_op.cc | 145 typename TTypes<T>::ConstScalar threshold, in operator ()()
185 typename TTypes<T>::ConstScalar threshold, \
|
| D | compare_and_bitpack_op.h | 34 typename TTypes<T>::ConstScalar threshold,
|
| D | scatter_functor.cc | 41 const typename TTypes<T>::ConstScalar update, \
|
| D | fill_functor.h | 32 typename TTypes<T>::ConstScalar in);
|
| D | fill_functor.cu.cc | 72 typename TTypes<T>::ConstScalar in) { in operator ()()
|
| D | constant_op_gpu.cu.cc | 72 typename TTypes<T>::ConstScalar in) { in operator ()()
|
| D | cwise_op_gpu_select.cu.cc | 40 typename TTypes<bool>::ConstScalar cond, in operator ()()
|
| D | cwise_op_select.cc | 219 TTypes<bool>::ConstScalar cond_scalar = cond->scalar<bool>(); in operator ()()
246 TTypes<bool>::ConstScalar cond, in operator ()()
|
| D | compare_and_bitpack_op_gpu.cu.cc | 121 typename TTypes<T>::ConstScalar threshold, \
|
| D | scatter_functor_gpu.cu.h | 142 const typename TTypes<T>::ConstScalar update,
|
| D | cwise_ops.h | 676 typedef typename TTypes<in_type>::ConstScalar tscalar_type;
1195 typename TTypes<bool>::ConstScalar cond,
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | tensor_types.h | 53 ConstScalar; typedef
|
| D | tensor.h | 513 typename TTypes<T>::ConstScalar scalar() const;
873 typename TTypes<T>::ConstScalar Tensor::scalar() const { in scalar()
875 return typename TTypes<T>::ConstScalar(base<T>()); in scalar()
|
| /external/tensorflow/tensorflow/contrib/seq2seq/kernels/ |
| D | beam_search_ops.cc | 87 typename TTypes<T>::ConstScalar end_token_t(end_token.scalar<T>()); in Compute()
|