| /external/tensorflow/tensorflow/core/kernels/ |
| D | training_ops.h | 34 typename TTypes<T>::ConstScalar alpha,
43 typename TTypes<T>::ConstScalar lr,
44 typename TTypes<T>::ConstScalar rho,
45 typename TTypes<T>::ConstScalar epsilon,
52 typename TTypes<T>::ConstScalar lr,
53 typename TTypes<T>::ConstScalar l1,
54 typename TTypes<T>::ConstScalar l2,
61 typename TTypes<T>::ConstScalar lr,
62 typename TTypes<T>::ConstScalar l1,
63 typename TTypes<T>::ConstScalar l2,
[all …]
|
| D | training_ops_gpu.cu.cc | 339 typename TTypes<T>::ConstScalar lr, in operator ()()
462 typename TTypes<T>::ConstScalar lr, in operator ()()
483 typename TTypes<T>::ConstScalar lr, in operator ()()
484 typename TTypes<T>::ConstScalar epsilon, in operator ()()
507 typename TTypes<T>::ConstScalar lr, in operator ()()
508 typename TTypes<T>::ConstScalar epsilon, in operator ()()
531 typename TTypes<T>::ConstScalar lr, in operator ()()
532 typename TTypes<T>::ConstScalar l1, in operator ()()
533 typename TTypes<T>::ConstScalar l2, in operator ()()
567 typename TTypes<T>::ConstScalar lr, in operator ()()
[all …]
|
| D | one_hot_op.h | 41 const typename TTypes<T>::ConstScalar& on_value, in OneGenerator()
42 const typename TTypes<T>::ConstScalar& off_value) in OneGenerator()
54 const typename TTypes<T>::ConstScalar on_value_;
55 const typename TTypes<T>::ConstScalar off_value_;
66 const typename TTypes<T>::ConstScalar& on_value, in Compute()
67 const typename TTypes<T>::ConstScalar& off_value, in Compute()
78 const typename TTypes<T>::ConstScalar& on_value,
79 const typename TTypes<T>::ConstScalar& off_value,
|
| D | training_ops.cc | 49 typename TTypes<T>::ConstScalar lr, in operator ()()
60 typename TTypes<T>::ConstScalar lr, in operator ()()
61 typename TTypes<T>::ConstScalar rho, in operator ()()
62 typename TTypes<T>::ConstScalar epsilon, in operator ()()
77 typename TTypes<T>::ConstScalar lr, in operator ()()
78 typename TTypes<T>::ConstScalar l1, in operator ()()
79 typename TTypes<T>::ConstScalar l2, in operator ()()
106 typename TTypes<T>::ConstScalar lr, int64 global_step, in operator ()()
107 typename TTypes<T>::ConstScalar l1, in operator ()()
108 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 | quantize_and_dequantize_op_gpu.cu.cc | 60 typename TTypes<T>::ConstScalar input_min_tensor, in operator ()()
61 typename TTypes<T>::ConstScalar input_max_tensor, in operator ()()
|
| D | compare_and_bitpack_op.h | 34 typename TTypes<T>::ConstScalar threshold,
|
| D | one_hot_op.cc | 153 const typename TTypes<T>::ConstScalar& on_value, \
154 const typename TTypes<T>::ConstScalar& off_value, \
|
| D | quantize_and_dequantize_op.h | 67 typename TTypes<T>::ConstScalar input_min,
68 typename TTypes<T>::ConstScalar input_max,
278 typename TTypes<T>::ConstScalar input_min, in Compute()
279 typename TTypes<T>::ConstScalar input_max, in Compute()
|
| D | fake_quant_ops.cc | 285 typename TTypes<float>::ConstScalar min,
286 typename TTypes<float>::ConstScalar max, const int quant_min,
299 typename TTypes<float>::ConstScalar min,
300 typename TTypes<float>::ConstScalar max, const int quant_min,
|
| D | compare_and_bitpack_op.cc | 145 typename TTypes<T>::ConstScalar threshold, in operator ()()
185 typename TTypes<T>::ConstScalar threshold, \
|
| D | scatter_functor.h | 317 const typename TTypes<T>::ConstScalar update,
325 const typename TTypes<T>::ConstScalar update,
348 const typename TTypes<Variant>::ConstScalar update,
385 const typename TTypes<T>::ConstScalar update,
|
| D | fill_functor.h | 32 typename TTypes<T>::ConstScalar in);
|
| D | scatter_functor.cc | 41 const typename TTypes<T>::ConstScalar update, \
|
| D | constant_op_gpu.cu.cc | 73 typename TTypes<T>::ConstScalar in) { in operator ()()
|
| D | fill_functor.cc | 98 typename TTypes<T>::ConstScalar in) { in operator ()()
|
| D | fill_functor.cu.cc | 73 typename TTypes<T>::ConstScalar in) { in operator ()()
|
| D | cwise_op_gpu_select.cu.cc | 56 typename TTypes<bool>::ConstScalar cond, in operator ()()
|
| D | compare_and_bitpack_op_gpu.cu.cc | 119 typename TTypes<T>::ConstScalar threshold, \
|
| D | quantize_and_dequantize_op.cc | 376 typename TTypes<T>::ConstScalar input_min_tensor, in operator ()()
377 typename TTypes<T>::ConstScalar input_max_tensor, in operator ()()
|
| /external/tensorflow/tensorflow/core/kernels/image/ |
| 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()
395 typename TTypes<float>::ConstScalar contrast_factor, \
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | tensor_types.h | 53 ConstScalar; typedef
|
| /external/tensorflow/tensorflow/core/kernels/sparse/ |
| D | mul_op.cc | 129 typename TTypes<T>::ConstScalar b, CSRSparseMatrix* c) { in Compute()
|
| D | kernels.h | 236 typename TTypes<T>::ConstScalar b, CSRSparseMatrix* c);
|