Home
last modified time | relevance | path

Searched refs:ConstFlat (Results 1 – 25 of 54) sorted by relevance

123

/external/tensorflow/tensorflow/core/kernels/
Daggregate_ops.h30 typename TTypes<T>::ConstFlat in1,
31 typename TTypes<T>::ConstFlat in2);
37 typename TTypes<T>::ConstFlat in1, in Compute()
38 typename TTypes<T>::ConstFlat in2) { in Compute()
46 typename TTypes<T>::ConstFlat in1,
47 typename TTypes<T>::ConstFlat in2,
48 typename TTypes<T>::ConstFlat in3);
54 typename TTypes<T>::ConstFlat in1, in Compute()
55 typename TTypes<T>::ConstFlat in2, in Compute()
56 typename TTypes<T>::ConstFlat in3) { in Compute()
[all …]
Daggregate_ops_cpu.h38 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 …]
Daggregate_ops_gpu.cu.cc35 typename TTypes<T>::ConstFlat in1, in operator ()()
36 typename TTypes<T>::ConstFlat in2) { in operator ()()
44 typename TTypes<T>::ConstFlat in1, in operator ()()
45 typename TTypes<T>::ConstFlat in2, in operator ()()
46 typename TTypes<T>::ConstFlat in3) { in operator ()()
54 typename TTypes<T>::ConstFlat in1, in operator ()()
55 typename TTypes<T>::ConstFlat in2, in operator ()()
56 typename TTypes<T>::ConstFlat in3, in operator ()()
57 typename TTypes<T>::ConstFlat in4) { in operator ()()
65 typename TTypes<T>::ConstFlat in1, in operator ()()
[all …]
Dcwise_op_clip.h26 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 …]
Dcwise_op_clip_gpu.cu.cc59 void operator()(const GPUDevice &d, typename TTypes<T>::ConstFlat &in0_flat, in operator ()()
60 typename TTypes<T>::ConstFlat &in1_flat, in operator ()()
61 typename TTypes<T>::ConstFlat &in2_flat, in operator ()()
75 void operator()(const GPUDevice &d, typename TTypes<T>::ConstFlat &in0_flat, in operator ()()
76 typename TTypes<T>::ConstFlat &in1_flat, in operator ()()
77 typename TTypes<T>::ConstFlat &in2_flat, in operator ()()
91 void operator()(const GPUDevice &d, typename TTypes<T>::ConstFlat &in0_flat, in operator ()()
92 typename TTypes<T>::ConstFlat &in1_flat, in operator ()()
93 typename TTypes<T>::ConstFlat &in2_flat, in operator ()()
107 void operator()(const GPUDevice &d, typename TTypes<T>::ConstFlat &in0_flat, in operator ()()
[all …]
Dcwise_op_clip.cc92 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 …]
Dtraining_ops.h34 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, bool update_slots);
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 …]
Ddense_update_functor.h41 typename TTypes<T>::ConstFlat update);
47 typename TTypes<T>::ConstFlat update) {
55 typename TTypes<T>::ConstFlat update) {
63 typename TTypes<T>::ConstFlat update) {
72 typename TTypes<T>::ConstFlat update) {
80 typename TTypes<T>::ConstFlat update) {
88 typename TTypes<T>::ConstFlat update) {
Dparameterized_truncated_normal_op.h39 typename TTypes<T>::ConstFlat means,
40 typename TTypes<T>::ConstFlat stddevs,
41 typename TTypes<T>::ConstFlat minvals,
42 typename TTypes<T>::ConstFlat maxvals,
Dcwise_op_gpu_select.cu.cc29 typename TTypes<bool>::ConstFlat cond_flat, in operator ()()
30 typename TTypes<T>::ConstFlat then_flat, in operator ()()
31 typename TTypes<T>::ConstFlat else_flat) { in operator ()()
41 typename TTypes<T>::ConstFlat then_flat, in operator ()()
42 typename TTypes<T>::ConstFlat else_flat) { in operator ()()
Dfake_quant_ops_functor.h72 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()
Ddata_format_ops.h29 void operator()(const Device& d, typename TTypes<T>::ConstFlat x, in operator()
56 typename TTypes<T>::ConstFlat input) const { in dimensions()
62 void eval(typename TTypes<T>::ConstFlat input, Output& output, in eval()
76 void operator()(const Device& d, typename TTypes<T>::ConstFlat x, in operator()
Dtraining_ops_gpu.cu.cc32 typename TTypes<T>::ConstFlat grad) { in operator ()()
45 typename TTypes<T>::ConstFlat grad, bool update_slots) { in operator ()()
64 typename TTypes<T>::ConstFlat grad) { in operator ()()
88 typename TTypes<T>::ConstFlat grad, in operator ()()
109 typename TTypes<T>::ConstFlat grad, in operator ()()
135 typename TTypes<T>::ConstFlat grad, bool use_nesterov) { in operator ()()
179 typename TTypes<T>::ConstFlat grad) { in operator ()()
210 typename TTypes<T>::ConstFlat grad) { in operator ()()
235 typename TTypes<T>::ConstFlat grad) { in operator ()()
260 typename TTypes<T>::ConstFlat grad) { in operator ()()
[all …]
Ddense_update_functor_gpu.cu.cc33 typename TTypes<T>::ConstFlat update) { in operator ()()
41 typename TTypes<T>::ConstFlat update) { in operator ()()
49 typename TTypes<T>::ConstFlat update) { in operator ()()
Dgather_functor.h41 typename TTypes<Index>::ConstFlat indices, in HandleCopies()
116 typename TTypes<Index>::ConstFlat indices, in operator()
153 typename TTypes<Index>::ConstFlat indices,
161 typename TTypes<Index>::ConstFlat indices,
171 typename TTypes<Index>::ConstFlat indices,
Dfake_quant_ops.cc152 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,
Dscatter_functor.h196 typename TTypes<Index>::ConstFlat indices);
204 typename TTypes<Index>::ConstFlat indices) {
227 typename TTypes<Index>::ConstFlat indices) {
264 typename TTypes<Index>::ConstFlat indices) {
288 typename TTypes<Index>::ConstFlat indices) {
352 typename TTypes<Index>::ConstFlat indices);
360 typename TTypes<Index>::ConstFlat indices) {
383 typename TTypes<Index>::ConstFlat indices) {
419 typename TTypes<Index>::ConstFlat indices) {
444 typename TTypes<Index>::ConstFlat indices) {
Dparameterized_truncated_normal_op_gpu.cu.cc237 typename TTypes<T>::ConstFlat means, in operator ()()
238 typename TTypes<T>::ConstFlat stddevs, in operator ()()
239 typename TTypes<T>::ConstFlat minvals, in operator ()()
240 typename TTypes<T>::ConstFlat maxvals, in operator ()()
Dcwise_op_select.cc195 typename TTypes<bool>::ConstFlat cond_flat, in operator ()()
196 typename TTypes<T>::ConstFlat then_flat, in operator ()()
197 typename TTypes<T>::ConstFlat else_flat) { in operator ()()
247 typename TTypes<T>::ConstFlat then_flat, in operator ()()
248 typename TTypes<T>::ConstFlat else_flat) { in operator ()()
Dcast_op.h35 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);
Dscatter_functor.cc35 typename TTypes<Index>::ConstFlat indices); \
42 typename TTypes<Index>::ConstFlat indices); \
Dragged_tensor_to_sparse_kernel.cc40 std::vector<TTypes<int64>::ConstFlat> rt_nested_splits; in Compute()
153 std::vector<TTypes<int64>::ConstFlat> rt_nested_splits, in ValidateInputs()
209 const std::vector<TTypes<int64>::ConstFlat>& rt_nested_splits) { in IsCompleted()
Dsegment_reduction_ops.h66 typename TTypes<Index>::ConstFlat segment_ids,
78 typename TTypes<Index>::ConstFlat segment_ids,
Dpopulation_count_op.cc98 void operator()(OpKernelContext* c, typename TTypes<T>::ConstFlat input, in operator ()()
146 OpKernelContext* c, typename TTypes<T>::ConstFlat input, \
Dparameterized_truncated_normal_op.cc54 typename TTypes<T>::ConstFlat means, in operator ()()
55 typename TTypes<T>::ConstFlat stddevs, in operator ()()
56 typename TTypes<T>::ConstFlat minvals, in operator ()()
57 typename TTypes<T>::ConstFlat maxvals, in operator ()()

123