| /external/tensorflow/tensorflow/core/kernels/ |
| D | aggregate_ops.h | 31 typename TTypes<T>::ConstFlat in1,
32 typename TTypes<T>::ConstFlat in2);
38 typename TTypes<T>::ConstFlat in1, in Compute()
39 typename TTypes<T>::ConstFlat in2) { in Compute()
47 typename TTypes<T>::ConstFlat in1,
48 typename TTypes<T>::ConstFlat in2,
49 typename TTypes<T>::ConstFlat in3);
55 typename TTypes<T>::ConstFlat in1, in Compute()
56 typename TTypes<T>::ConstFlat in2, in Compute()
57 typename TTypes<T>::ConstFlat in3) { in Compute()
[all …]
|
| D | aggregate_ops_gpu.cu.cc | 36 typename TTypes<T>::ConstFlat in1, in operator ()()
37 typename TTypes<T>::ConstFlat in2) { in operator ()()
45 typename TTypes<T>::ConstFlat in1, in operator ()()
46 typename TTypes<T>::ConstFlat in2, in operator ()()
47 typename TTypes<T>::ConstFlat in3) { in operator ()()
55 typename TTypes<T>::ConstFlat in1, in operator ()()
56 typename TTypes<T>::ConstFlat in2, in operator ()()
57 typename TTypes<T>::ConstFlat in3, in operator ()()
58 typename TTypes<T>::ConstFlat in4) { in operator ()()
66 typename TTypes<T>::ConstFlat in1, in operator ()()
[all …]
|
| D | aggregate_ops_cpu.h | 35 typename TTypes<T>::ConstFlat in1,
36 typename TTypes<T>::ConstFlat in2) {
43 typename TTypes<T>::ConstFlat in1,
44 typename TTypes<T>::ConstFlat in2,
45 typename TTypes<T>::ConstFlat in3) {
52 typename TTypes<T>::ConstFlat in1,
53 typename TTypes<T>::ConstFlat in2,
54 typename TTypes<T>::ConstFlat in3,
55 typename TTypes<T>::ConstFlat in4) {
62 typename TTypes<T>::ConstFlat in1,
[all …]
|
| D | cwise_op_clip.h | 26 void operator()(const Device &d, typename TTypes<T>::ConstFlat &in0_flat,
27 typename TTypes<T>::ConstFlat &in1_flat,
28 typename TTypes<T>::ConstFlat &in2_flat,
35 void operator()(const Device &d, typename TTypes<T>::ConstFlat &in0_flat,
36 typename TTypes<T>::ConstFlat &in1_flat,
37 typename TTypes<T>::ConstFlat &in2_flat,
44 void operator()(const Device &d, typename TTypes<T>::ConstFlat &in0_flat,
45 typename TTypes<T>::ConstFlat &in1_flat,
46 typename TTypes<T>::ConstFlat &in2_flat,
53 void operator()(const Device &d, typename TTypes<T>::ConstFlat &in0_flat,
[all …]
|
| D | parameterized_truncated_normal_op.h | 40 typename TTypes<T>::ConstFlat means,
41 typename TTypes<T>::ConstFlat stddevs,
42 typename TTypes<T>::ConstFlat minvals,
43 typename TTypes<T>::ConstFlat maxvals,
55 typename TTypes<T>::ConstFlat means,
56 typename TTypes<T>::ConstFlat stddevs,
57 typename TTypes<T>::ConstFlat minvals,
58 typename TTypes<T>::ConstFlat maxvals,
|
| D | cwise_op_clip_gpu.cu.cc | 67 void operator()(const GPUDevice &d, typename TTypes<T>::ConstFlat &in0_flat, in operator ()()
68 typename TTypes<T>::ConstFlat &in1_flat, in operator ()()
69 typename TTypes<T>::ConstFlat &in2_flat, in operator ()()
83 void operator()(const GPUDevice &d, typename TTypes<T>::ConstFlat &in0_flat, in operator ()()
84 typename TTypes<T>::ConstFlat &in1_flat, in operator ()()
85 typename TTypes<T>::ConstFlat &in2_flat, in operator ()()
99 void operator()(const GPUDevice &d, typename TTypes<T>::ConstFlat &in0_flat, in operator ()()
100 typename TTypes<T>::ConstFlat &in1_flat, in operator ()()
101 typename TTypes<T>::ConstFlat &in2_flat, in operator ()()
115 void operator()(const GPUDevice &d, typename TTypes<T>::ConstFlat &in0_flat, in operator ()()
[all …]
|
| D | cwise_op_clip.cc | 92 void operator()(const CPUDevice& d, typename TTypes<T>::ConstFlat& in0_flat, in operator ()()
93 typename TTypes<T>::ConstFlat& in1_flat, in operator ()()
94 typename TTypes<T>::ConstFlat& in2_flat, in operator ()()
111 void operator()(const CPUDevice& d, typename TTypes<T>::ConstFlat& in0_flat, in operator ()()
112 typename TTypes<T>::ConstFlat& in1_flat, in operator ()()
113 typename TTypes<T>::ConstFlat& in2_flat, in operator ()()
131 void operator()(const CPUDevice& d, typename TTypes<T>::ConstFlat& in0_flat, in operator ()()
132 typename TTypes<T>::ConstFlat& in1_flat, in operator ()()
133 typename TTypes<T>::ConstFlat& in2_flat, in operator ()()
143 void operator()(const CPUDevice& d, typename TTypes<T>::ConstFlat& in0_flat, in operator ()()
[all …]
|
| D | training_ops.h | 35 typename TTypes<T>::ConstFlat delta);
46 typename TTypes<T>::ConstFlat grad);
55 typename TTypes<T>::ConstFlat grad);
64 typename TTypes<T>::ConstFlat grad);
72 typename TTypes<T>::ConstFlat grad, bool update_slots);
81 typename TTypes<T>::ConstFlat grad, bool update_slots);
92 typename TTypes<T>::ConstFlat grad);
114 typename TTypes<T>::ConstFlat grad);
133 typename TTypes<T>::ConstFlat grad,
145 typename TTypes<T>::ConstFlat grad,
[all …]
|
| D | dense_update_functor.h | 38 typename TTypes<T>::ConstFlat update);
44 typename TTypes<T>::ConstFlat update) {
52 typename TTypes<T>::ConstFlat update) {
60 typename TTypes<T>::ConstFlat update) {
|
| D | cwise_op_gpu_select.cu.cc | 45 typename TTypes<bool>::ConstFlat cond_flat, in operator ()()
46 typename TTypes<T>::ConstFlat then_flat, in operator ()()
47 typename TTypes<T>::ConstFlat else_flat) { in operator ()()
57 typename TTypes<T>::ConstFlat then_flat, in operator ()()
58 typename TTypes<T>::ConstFlat else_flat) { in operator ()()
|
| D | fake_quant_ops_functor.h | 72 using ConstFlat = typename tensorflow::TTypes<T>::ConstFlat; variable
80 void operator()(const Device& d, ConstFlat<float> inputs, const float min, in operator()
104 void operator()(const Device& d, ConstFlat<float> gradients, in operator()
105 ConstFlat<float> inputs, const float min, const float max, in operator()
127 void operator()(const Device& d, ConstFlat<float> inputs, in operator()
155 void operator()(const Device& d, ConstFlat<float> gradients, in operator()
156 ConstFlat<float> inputs, ConstScalar<float> min, in operator()
|
| D | dense_update_functor_gpu.cu.cc | 33 typename TTypes<T>::ConstFlat update) { in operator ()()
41 typename TTypes<T>::ConstFlat update) { in operator ()()
49 typename TTypes<T>::ConstFlat update) { in operator ()()
|
| D | data_format_ops.h | 29 void operator()(const Device& d, typename TTypes<T>::ConstFlat x, in operator()
82 typename TTypes<T>::ConstFlat input) const { in dimensions()
88 void eval(typename TTypes<T>::ConstFlat input, Output& output, in eval()
102 void operator()(const Device& d, typename TTypes<T>::ConstFlat x, in operator()
|
| D | gather_functor.h | 41 typename TTypes<Index>::ConstFlat indices, in HandleCopies()
117 typename TTypes<Index>::ConstFlat indices, in operator()
158 typename TTypes<Index>::ConstFlat indices,
166 typename TTypes<Index>::ConstFlat indices,
176 typename TTypes<Index>::ConstFlat indices,
|
| D | gather_functor_batched.h | 41 typename TTypes<Index>::ConstFlat indices, in HandleCopiesBatched()
131 typename TTypes<Index>::ConstFlat indices, in operator()
175 typename TTypes<Index>::ConstFlat indices,
183 typename TTypes<Index>::ConstFlat indices,
193 typename TTypes<Index>::ConstFlat indices,
|
| D | scatter_functor.h | 135 typename TTypes<Index>::ConstFlat indices);
143 typename TTypes<Index>::ConstFlat indices) {
184 typename TTypes<Index>::ConstFlat indices) {
204 typename TTypes<Index>::ConstFlat indices) {
238 typename TTypes<Index>::ConstFlat indices) {
275 typename TTypes<Index>::ConstFlat indices) {
318 typename TTypes<Index>::ConstFlat indices);
326 typename TTypes<Index>::ConstFlat indices) {
349 typename TTypes<Index>::ConstFlat indices) {
386 typename TTypes<Index>::ConstFlat indices) {
|
| D | fake_quant_ops.cc | 163 const GPUDevice& d, typename TTypes<float>::ConstFlat inputs,
172 const GPUDevice& d, typename TTypes<float>::ConstFlat gradients,
173 typename TTypes<float>::ConstFlat inputs, const float min, const float max,
284 const GPUDevice& d, typename TTypes<float>::ConstFlat inputs,
297 const GPUDevice& d, typename TTypes<float>::ConstFlat gradients,
298 typename TTypes<float>::ConstFlat inputs,
433 typename TTypes<float>::ConstFlat min,
434 typename TTypes<float>::ConstFlat max, const int quant_min,
|
| D | random_binomial_op.h | 52 typename TTypes<T>::ConstFlat counts,
53 typename TTypes<T>::ConstFlat probs,
|
| D | cast_op.h | 35 typename TTypes<IN_OUT>::ConstFlat in_tensor, \
53 typename TTypes<IN_OUT>::ConstFlat in_tensor, \
80 typename TTypes<IN_OUT>::ConstFlat in_tensor, \
141 typename TTypes<Tin>::ConstFlat i) { in Cast()
148 typename TTypes<Tin>::ConstFlat i, bool truncate = false);
|
| D | scatter_functor.cc | 35 typename TTypes<Index>::ConstFlat indices); \
42 typename TTypes<Index>::ConstFlat indices); \
|
| D | parameterized_truncated_normal_op_gpu.cu.cc | 238 typename TTypes<T>::ConstFlat means, in operator ()()
239 typename TTypes<T>::ConstFlat stddevs, in operator ()()
240 typename TTypes<T>::ConstFlat minvals, in operator ()()
241 typename TTypes<T>::ConstFlat maxvals, in operator ()()
|
| D | quantize_and_dequantize_op_gpu.cu.cc | 58 void operator()(const GPUDevice& d, typename TTypes<T>::ConstFlat gradient, in operator ()()
59 typename TTypes<T>::ConstFlat input, in operator ()()
|
| D | parameterized_truncated_normal_op.cc | 55 typename TTypes<T>::ConstFlat means, in operator ()()
56 typename TTypes<T>::ConstFlat stddevs, in operator ()()
57 typename TTypes<T>::ConstFlat minvals, in operator ()()
58 typename TTypes<T>::ConstFlat maxvals, in operator ()()
318 typename TTypes<T>::ConstFlat means, in operator ()()
319 typename TTypes<T>::ConstFlat stddevs, in operator ()()
320 typename TTypes<T>::ConstFlat minvals, in operator ()()
321 typename TTypes<T>::ConstFlat maxvals, in operator ()()
|
| D | training_ops_gpu.cu.cc | 340 typename TTypes<T>::ConstFlat grad) { in operator ()()
463 typename TTypes<T>::ConstFlat grad, bool update_slots) { in operator ()()
485 typename TTypes<T>::ConstFlat grad, bool update_slots) { in operator ()()
534 typename TTypes<T>::ConstFlat grad) { in operator ()()
596 typename TTypes<T>::ConstFlat grad) { in operator ()()
625 typename TTypes<T>::ConstFlat grad, in operator ()()
660 typename TTypes<T>::ConstFlat grad, in operator ()()
695 typename TTypes<T>::ConstFlat grad, in operator ()()
735 typename TTypes<T>::ConstFlat grad, in operator ()()
813 typename TTypes<T>::ConstFlat grad, in operator ()()
[all …]
|
| D | segment_reduction_ops.h | 54 typename TTypes<Index>::ConstFlat segment_ids,
65 typename TTypes<Index>::ConstFlat segment_ids,
|