| /external/tensorflow/tensorflow/core/kernels/ |
| aggregate_ops.h | 30 typename TTypes<T>::ConstFlat in1,
31 typename TTypes<T>::ConstFlat in2);
37 typename TTypes<T>::ConstFlat in1,
38 typename TTypes<T>::ConstFlat in2) {
46 typename TTypes<T>::ConstFlat in1,
47 typename TTypes<T>::ConstFlat in2,
48 typename TTypes<T>::ConstFlat in3);
54 typename TTypes<T>::ConstFlat in1,
55 typename TTypes<T>::ConstFlat in2,
56 typename TTypes<T>::ConstFlat in3)
[all...] |
| aggregate_ops_cpu.h | 38 typename TTypes<T>::ConstFlat in1,
39 typename TTypes<T>::ConstFlat in2) {
46 typename TTypes<T>::ConstFlat in1,
47 typename TTypes<T>::ConstFlat in2,
48 typename TTypes<T>::ConstFlat in3) {
55 typename TTypes<T>::ConstFlat in1,
56 typename TTypes<T>::ConstFlat in2,
57 typename TTypes<T>::ConstFlat in3,
58 typename TTypes<T>::ConstFlat in4) {
65 typename TTypes<T>::ConstFlat in1
[all...] |
| aggregate_ops_gpu.cu.cc | 35 typename TTypes<T>::ConstFlat in1,
36 typename TTypes<T>::ConstFlat in2) {
44 typename TTypes<T>::ConstFlat in1,
45 typename TTypes<T>::ConstFlat in2,
46 typename TTypes<T>::ConstFlat in3) {
54 typename TTypes<T>::ConstFlat in1,
55 typename TTypes<T>::ConstFlat in2,
56 typename TTypes<T>::ConstFlat in3,
57 typename TTypes<T>::ConstFlat in4) {
65 typename TTypes<T>::ConstFlat in1
[all...] |
| parameterized_truncated_normal_op.h | 39 typename TTypes<T>::ConstFlat means,
40 typename TTypes<T>::ConstFlat stddevs,
41 typename TTypes<T>::ConstFlat minvals,
42 typename TTypes<T>::ConstFlat maxvals,
|
| population_count_op.h | 30 void operator()(OpKernelContext* c, typename TTypes<T>::ConstFlat input,
|
| 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) {
69 typename TTypes<T>::ConstFlat update) {
77 typename TTypes<T>::ConstFlat update) {
85 typename TTypes<T>::ConstFlat update) {
|
| sparse_tensor_dense_add_op.h | 35 typename TTypes<T>::ConstFlat updates,
|
| dense_update_functor_gpu.cu.cc | 33 typename TTypes<T>::ConstFlat update) {
41 typename TTypes<T>::ConstFlat update) {
49 typename TTypes<T>::ConstFlat update) {
|
| training_ops.h | 34 typename TTypes<T>::ConstFlat delta);
45 typename TTypes<T>::ConstFlat grad);
54 typename TTypes<T>::ConstFlat grad);
63 typename TTypes<T>::ConstFlat grad);
71 typename TTypes<T>::ConstFlat grad);
82 typename TTypes<T>::ConstFlat grad);
92 typename TTypes<T>::ConstFlat grad);
100 typename TTypes<T>::ConstFlat grad,
112 typename TTypes<T>::ConstFlat grad,
125 typename TTypes<T>::ConstFlat grad
[all...] |
| cwise_op_gpu_select.cu.cc | 29 typename TTypes<bool>::ConstFlat cond_flat,
30 typename TTypes<T>::ConstFlat then_flat,
31 typename TTypes<T>::ConstFlat else_flat) {
41 typename TTypes<T>::ConstFlat then_flat,
42 typename TTypes<T>::ConstFlat else_flat) {
|
| data_format_ops.h | 29 void operator()(const Device& d, typename TTypes<T>::ConstFlat x,
45 typename TTypes<T>::ConstFlat input) const {
51 void eval(typename TTypes<T>::ConstFlat input, Output& output,
74 typename TTypes<T>::ConstFlat input) const {
80 void eval(typename TTypes<T>::ConstFlat input, Output& output,
103 void operator()(const Device& d, typename TTypes<T>::ConstFlat x,
|
| gather_functor.cc | 32 typename TTypes<Index>::ConstFlat Tindices, \
|
| where_op.h | 44 typename TTypes<T>::ConstFlat input,
|
| fake_quant_ops_functor.h | 72 using ConstFlat = typename tensorflow::TTypes<T>::ConstFlat;
80 void operator()(const Device& d, ConstFlat<float> inputs, const float min,
104 void operator()(const Device& d, ConstFlat<float> gradients,
105 ConstFlat<float> inputs, const float min, const float max,
127 void operator()(const Device& d, ConstFlat<float> inputs,
155 void operator()(const Device& d, ConstFlat<float> gradients,
156 ConstFlat<float> inputs, ConstScalar<float> min,
|
| segment_reduction_ops.h | 45 typename TTypes<Index>::ConstFlat segment_ids,
57 typename TTypes<Index>::ConstFlat segment_ids,
|
| fake_quant_ops.cc | 152 const GPUDevice& d, typename TTypes<float>::ConstFlat inputs,
161 const GPUDevice& d, typename TTypes<float>::ConstFlat gradients,
162 typename TTypes<float>::ConstFlat inputs, const float min, const float max,
271 const GPUDevice& d, typename TTypes<float>::ConstFlat inputs,
284 const GPUDevice& d, typename TTypes<float>::ConstFlat gradients,
285 typename TTypes<float>::ConstFlat inputs,
418 typename TTypes<float>::ConstFlat min,
419 typename TTypes<float>::ConstFlat max, const int quant_min,
|
| cast_op_gpu.cu.cc | 31 typename TTypes<I>::ConstFlat i) {
|
| dense_update_functor.cc | 35 typename TTypes<string>::ConstFlat update) {
|
| gather_functor.h | 39 typename TTypes<Index>::ConstFlat indices,
114 typename TTypes<Index>::ConstFlat indices,
151 typename TTypes<Index>::ConstFlat indices,
159 typename TTypes<Index>::ConstFlat indices,
|
| parameterized_truncated_normal_op_gpu.cu.cc | 197 typename TTypes<T>::ConstFlat means,
198 typename TTypes<T>::ConstFlat stddevs,
199 typename TTypes<T>::ConstFlat minvals,
200 typename TTypes<T>::ConstFlat maxvals,
|
| scatter_functor.cc | 35 typename TTypes<Index>::ConstFlat indices); \
|
| string_join_op.cc | 42 std::vector<TTypes<string>::ConstFlat> inputs;
|
| cwise_op_select.cc | 196 typename TTypes<bool>::ConstFlat cond_flat,
197 typename TTypes<T>::ConstFlat then_flat,
198 typename TTypes<T>::ConstFlat else_flat) {
214 typename TTypes<T>::ConstFlat then_flat,
215 typename TTypes<T>::ConstFlat else_flat) {
|
| training_ops_gpu.cu.cc | 32 typename TTypes<T>::ConstFlat grad) {
45 typename TTypes<T>::ConstFlat grad) {
62 typename TTypes<T>::ConstFlat grad) {
86 typename TTypes<T>::ConstFlat grad,
112 typename TTypes<T>::ConstFlat grad, bool use_nesterov) {
153 typename TTypes<T>::ConstFlat grad) {
178 typename TTypes<T>::ConstFlat grad) {
202 typename TTypes<T>::ConstFlat grad) {
234 typename TTypes<T>::ConstFlat grad) {
|
| cast_op.h | 59 typename TTypes<Tin>::ConstFlat i) {
66 typename TTypes<Tin>::ConstFlat i);
|
