| /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 ()()
59 typename TTypes<T>::ConstScalar lr, in operator ()()
60 typename TTypes<T>::ConstScalar rho, in operator ()()
61 typename TTypes<T>::ConstScalar epsilon, in operator ()()
85 typename TTypes<T>::ConstScalar lr, in operator ()()
87 typename TTypes<T>::ConstScalar momentum, bool use_nesterov) { in operator ()()
106 typename TTypes<T>::ConstScalar beta1_power, in operator ()()
107 typename TTypes<T>::ConstScalar beta2_power, in operator ()()
108 typename TTypes<T>::ConstScalar lr, 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 | 50 typename TTypes<T>::ConstScalar lr, in operator ()()
71 typename TTypes<T>::ConstScalar lr, in operator ()()
72 typename TTypes<T>::ConstScalar rho, in operator ()()
73 typename TTypes<T>::ConstScalar epsilon, in operator ()()
88 typename TTypes<T>::ConstScalar lr, in operator ()()
89 typename TTypes<T>::ConstScalar l1, in operator ()()
90 typename TTypes<T>::ConstScalar l2, in operator ()()
117 typename TTypes<T>::ConstScalar lr, int64 global_step, in operator ()()
118 typename TTypes<T>::ConstScalar l1, in operator ()()
119 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, \
369 typename TTypes<float>::ConstScalar factor, in IncrementWithScaling()
390 typename TTypes<float>::ConstScalar contrast_factor,
|
| 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 | fill_functor.cc | 123 typename TTypes<T>::ConstScalar in) { in operator ()()
141 typename TTypes<T>::ConstScalar in) { in operator ()()
|
| 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 | cwise_op_select.cc | 139 TTypes<bool>::ConstScalar cond_scalar = cond->scalar<bool>(); in ComputeScalar()
213 TTypes<bool>::ConstScalar cond, in operator ()()
|
| 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 | compare_and_bitpack_op_gpu.cu.cc | 121 typename TTypes<T>::ConstScalar threshold, \
|
| D | cwise_ops.h | 510 typedef typename TTypes<in_type>::ConstScalar tscalar_type;
1009 typename TTypes<bool>::ConstScalar cond,
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | tensor.h | 418 typename TTypes<T>::ConstScalar scalar() const;
722 typename TTypes<T>::ConstScalar Tensor::scalar() const { in scalar()
724 return typename TTypes<T>::ConstScalar(base<T>()); in scalar()
|
| D | tensor_types.h | 53 ConstScalar; typedef
|
| /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()
|