Home
last modified time | relevance | path

Searched refs:TTypes (Results 1 – 25 of 280) sorted by relevance

12345678910>>...12

/external/tensorflow/tensorflow/core/kernels/
Dtraining_ops.h32 void operator()(const Device& d, typename TTypes<T>::Flat var,
33 typename TTypes<T>::ConstScalar alpha,
34 typename TTypes<T>::ConstFlat delta);
39 void operator()(const Device& d, typename TTypes<T>::Flat var,
40 typename TTypes<T>::Flat accum,
41 typename TTypes<T>::Flat accum_update,
42 typename TTypes<T>::ConstScalar lr,
43 typename TTypes<T>::ConstScalar rho,
44 typename TTypes<T>::ConstScalar epsilon,
45 typename TTypes<T>::ConstFlat grad);
[all …]
Daggregate_ops.h29 void operator()(const Device& d, typename TTypes<T>::Flat out,
30 typename TTypes<T>::ConstFlat in1,
31 typename TTypes<T>::ConstFlat in2);
36 static void Compute(const Device& d, typename TTypes<T>::Flat out, in Compute()
37 typename TTypes<T>::ConstFlat in1, in Compute()
38 typename TTypes<T>::ConstFlat in2) { in Compute()
45 void operator()(const Device& d, typename TTypes<T>::Flat out,
46 typename TTypes<T>::ConstFlat in1,
47 typename TTypes<T>::ConstFlat in2,
48 typename TTypes<T>::ConstFlat in3);
[all …]
Daggregate_ops_cpu.h37 void operator()(const CPUDevice& d, typename TTypes<T>::Flat out,
38 typename TTypes<T>::ConstFlat in1,
39 typename TTypes<T>::ConstFlat in2) {
45 void operator()(const CPUDevice& d, typename TTypes<T>::Flat out,
46 typename TTypes<T>::ConstFlat in1,
47 typename TTypes<T>::ConstFlat in2,
48 typename TTypes<T>::ConstFlat in3) {
54 void operator()(const CPUDevice& d, typename TTypes<T>::Flat out,
55 typename TTypes<T>::ConstFlat in1,
56 typename TTypes<T>::ConstFlat in2,
[all …]
Daggregate_ops_gpu.cu.cc34 void operator()(const GPUDevice& d, typename TTypes<T>::Flat out, in operator ()()
35 typename TTypes<T>::ConstFlat in1, in operator ()()
36 typename TTypes<T>::ConstFlat in2) { in operator ()()
43 void operator()(const GPUDevice& d, typename TTypes<T>::Flat out, 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 ()()
53 void operator()(const GPUDevice& d, typename TTypes<T>::Flat out, in operator ()()
54 typename TTypes<T>::ConstFlat in1, in operator ()()
55 typename TTypes<T>::ConstFlat in2, in operator ()()
[all …]
Dtraining_ops_gpu.cu.cc30 void operator()(const GPUDevice& d, typename TTypes<T>::Flat var, in operator ()()
31 typename TTypes<T>::ConstScalar lr, in operator ()()
32 typename TTypes<T>::ConstFlat grad) { in operator ()()
33 Eigen::array<typename TTypes<T>::Tensor::Index, 1> bcast; in operator ()()
42 void operator()(const GPUDevice& d, typename TTypes<T>::Flat var, in operator ()()
43 typename TTypes<T>::Flat accum, in operator ()()
44 typename TTypes<T>::ConstScalar lr, in operator ()()
45 typename TTypes<T>::ConstFlat grad) { in operator ()()
47 Eigen::array<typename TTypes<T>::Tensor::Index, 1> bcast; in operator ()()
56 void operator()(const GPUDevice& d, typename TTypes<T>::Flat var, in operator ()()
[all …]
Dsparse_xent_op.h31 typename TTypes<const T, 1>::Tensor32Bit To32BitConst( in To32BitConst()
32 typename TTypes<T>::Vec in) { in To32BitConst()
33 return To32Bit(typename TTypes<T>::ConstVec(in.data(), in.dimensions())); in To32BitConst()
37 typename TTypes<const T, 2>::Tensor32Bit To32BitConst( in To32BitConst()
38 typename TTypes<T>::Matrix in) { in To32BitConst()
39 return To32Bit(typename TTypes<T>::ConstMatrix(in.data(), in.dimensions())); in To32BitConst()
59 typename TTypes<const T, 2>::Tensor32Bit logits, in SparseXentLossGenerator()
60 typename TTypes<const T, 1>::Tensor32Bit sum_exp_logits, in SparseXentLossGenerator()
61 typename TTypes<const Index, 1>::Tensor32Bit labels, in SparseXentLossGenerator()
82 typename TTypes<const T, 2>::Tensor32Bit logits_;
[all …]
Dcrop_and_resize_op.h31 typename TTypes<T, 4>::ConstTensor image,
32 typename TTypes<float, 2>::ConstTensor boxes,
33 typename TTypes<int32, 1>::ConstTensor box_ind,
35 typename TTypes<float, 4>::Tensor crops);
41 bool operator()(const Device& d, typename TTypes<float, 4>::ConstTensor grads,
42 typename TTypes<float, 2>::ConstTensor boxes,
43 typename TTypes<int32, 1>::ConstTensor box_ind,
44 typename TTypes<T, 4>::Tensor grads_image);
50 bool operator()(const Device& d, typename TTypes<float, 4>::ConstTensor grads,
51 typename TTypes<T, 4>::ConstTensor image,
[all …]
Drelu_op_functor.h33 void operator()(const Device& d, typename TTypes<T>::ConstTensor features, in operator()
34 typename TTypes<T>::Tensor activations) { in operator()
48 void operator()(const Device& d, typename TTypes<T>::ConstTensor gradients, in operator()
49 typename TTypes<T>::ConstTensor features, in operator()
50 typename TTypes<T>::Tensor backprops) { in operator()
66 void operator()(const Device& d, typename TTypes<T>::ConstTensor features, in operator()
67 typename TTypes<T>::Tensor activations) { in operator()
81 void operator()(const Device& d, typename TTypes<T>::ConstTensor gradients, in operator()
82 typename TTypes<T>::ConstTensor features, in operator()
83 typename TTypes<T>::Tensor backprops) { in operator()
[all …]
Dconv_2d.h34 typename TTypes<T, Dims, IndexType>::ConstTensor input, in operator()
37 typename TTypes<T, Dims, IndexType>::Tensor output) { in operator()
45 const Device& d, typename TTypes<T, Dims, IndexType>::ConstTensor input, in operator()
49 typename TTypes<T, Dims, IndexType>::Tensor output) { in operator()
67 void operator()(const Device& d, typename TTypes<T, 4>::Tensor output, in operator()
68 typename TTypes<T, 4>::ConstTensor input, in operator()
69 typename TTypes<T, 4>::ConstTensor filter, int row_stride, in operator()
80 typename TTypes<Eigen::half, 4>::Tensor output,
81 typename TTypes<Eigen::half, 4>::ConstTensor input,
82 typename TTypes<Eigen::half, 4>::ConstTensor filter,
[all …]
Drelu_op.cc77 const GPUDevice& d, typename TTypes<T>::ConstTensor features, \
78 typename TTypes<T>::Tensor activations); \
83 const GPUDevice& d, typename TTypes<T>::ConstTensor gradients, \
84 typename TTypes<T>::ConstTensor features, \
85 typename TTypes<T>::Tensor backprops); \
90 const GPUDevice& d, typename TTypes<T>::ConstTensor features, \
91 typename TTypes<T>::Tensor activations); \
96 const GPUDevice& d, typename TTypes<T>::ConstTensor gradients, \
97 typename TTypes<T>::ConstTensor features, \
98 typename TTypes<T>::Tensor backprops); \
[all …]
Dbatch_norm_op.h28 void operator()(const Device& d, typename TTypes<T, 4>::ConstTensor input, in operator()
29 typename TTypes<T>::ConstVec mean, in operator()
30 typename TTypes<T>::ConstVec var, in operator()
31 typename TTypes<T>::ConstVec beta, in operator()
32 typename TTypes<T>::ConstVec gamma, T variance_epsilon, in operator()
34 typename TTypes<T, 4>::Tensor output) { in operator()
75 void operator()(const Device& d, typename TTypes<T, 4>::ConstTensor input, in operator()
76 typename TTypes<T>::ConstVec mean, in operator()
77 typename TTypes<T>::ConstVec var, in operator()
78 typename TTypes<T>::ConstVec gamma, in operator()
[all …]
Ddense_update_functor.h37 void operator()(const Device& d, typename TTypes<T>::Flat params,
38 typename TTypes<T>::ConstFlat update);
43 void operator()(const CPUDevice& d, typename TTypes<T>::Flat params,
44 typename TTypes<T>::ConstFlat update) {
51 void operator()(const CPUDevice& d, typename TTypes<T>::Flat params,
52 typename TTypes<T>::ConstFlat update) {
59 void operator()(const CPUDevice& d, typename TTypes<T>::Flat params,
60 typename TTypes<T>::ConstFlat update) {
68 void operator()(const SYCLDevice& d, typename TTypes<T>::Flat params,
69 typename TTypes<T>::ConstFlat update) {
[all …]
Ddilation_ops.h29 void operator()(const Device& d, typename TTypes<T, 4>::ConstTensor input,
30 typename TTypes<T, 3>::ConstTensor filter, int stride_rows,
32 int pad_left, typename TTypes<T, 4>::Tensor output);
41 void operator()(const Device& d, typename TTypes<T, 4>::ConstTensor input,
42 typename TTypes<T, 3>::ConstTensor filter,
43 typename TTypes<T, 4>::ConstTensor out_backprop,
46 typename TTypes<T, 4>::Tensor in_backprop);
55 void operator()(const Device& d, typename TTypes<T, 4>::ConstTensor input,
56 typename TTypes<T, 3>::ConstTensor filter,
57 typename TTypes<T, 4>::ConstTensor out_backprop,
[all …]
Dfake_quant_ops.cc39 using tensorflow::TTypes; // NOLINT This is needed in CUDA mode, do not remove.
152 const GPUDevice& d, typename TTypes<float>::ConstFlat inputs,
154 typename TTypes<float>::Flat outputs);
161 const GPUDevice& d, typename TTypes<float>::ConstFlat gradients,
162 typename TTypes<float>::ConstFlat inputs, const float min, const float max,
164 typename TTypes<float>::Flat backprops);
271 const GPUDevice& d, typename TTypes<float>::ConstFlat inputs,
272 typename TTypes<float>::ConstScalar min,
273 typename TTypes<float>::ConstScalar max, const int quant_min,
274 const int quant_max, typename TTypes<float>::Flat output);
[all …]
Dcwise_op_gpu_select.cu.cc28 void operator()(const GPUDevice& d, typename TTypes<T>::Flat out, in operator ()()
29 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 ()()
39 void operator()(const GPUDevice& d, typename TTypes<T>::Flat out, in operator ()()
40 typename TTypes<bool>::ConstScalar cond, in operator ()()
41 typename TTypes<T>::ConstFlat then_flat, in operator ()()
42 typename TTypes<T>::ConstFlat else_flat) { in operator ()()
60 typename TTypes<T>::Matrix output_flat_outer_dims, in operator ()()
61 TTypes<bool>::ConstVec cond_vec, in operator ()()
[all …]
Dgather_nd_op_cpu_impl.h45 const Index slice_size, typename TTypes<Index>::ConstMatrix Tindices, in GatherNdSliceGenerator()
46 typename TTypes<T, IXDIM + 1>::ConstTensor Tparams, in GatherNdSliceGenerator()
47 typename TTypes<T>::Matrix Tout, std::atomic<Index>* error_loc) in GatherNdSliceGenerator()
86 const typename TTypes<Index>::ConstMatrix Tindices_;
87 const typename TTypes<T, IXDIM + 1>::ConstTensor Tparams_;
88 mutable typename TTypes<T>::Matrix Tout_;
99 typename TTypes<int32>::Scalar Tscratch,
100 typename TTypes<T, IXDIM + 1>::ConstTensor Tparams,
101 typename TTypes<Index>::ConstMatrix Tindices,
102 typename TTypes<T>::Matrix Tout) {
[all …]
Done_hot_op.h34 OneGenerator(const typename TTypes<TI>::ConstMatrix& indices, in OneGenerator()
35 const typename TTypes<T>::ConstScalar& on_value, in OneGenerator()
36 const typename TTypes<T>::ConstScalar& off_value) in OneGenerator()
47 const typename TTypes<TI>::ConstMatrix indices_;
48 const typename TTypes<T>::ConstScalar on_value_;
49 const typename TTypes<T>::ConstScalar off_value_;
59 const Device& d, const typename TTypes<TI>::ConstMatrix& indices, in Compute()
60 const typename TTypes<T>::ConstScalar& on_value, in Compute()
61 const typename TTypes<T>::ConstScalar& off_value, in Compute()
62 typename TTypes<T, 3>::Tensor* output) { in Compute()
Dbatch_norm_op.cc180 const GPUDevice& d, typename TTypes<T, 4>::ConstTensor input, \
181 typename TTypes<T>::ConstVec mean, typename TTypes<T>::ConstVec var, \
182 typename TTypes<T>::ConstVec beta, typename TTypes<T>::ConstVec gamma, \
184 typename TTypes<T, 4>::Tensor output); \
236 const GPUDevice& d, typename TTypes<T, 4>::ConstTensor input, \
237 typename TTypes<T>::ConstVec mean, typename TTypes<T>::ConstVec var, \
238 typename TTypes<T>::ConstVec gamma, \
239 typename TTypes<T, 4>::ConstTensor out_backprop, T variance_epsilon, \
240 bool scale_after_normalization, typename TTypes<T, 4>::Tensor dx, \
241 typename TTypes<T>::Vec dm, typename TTypes<T>::Vec dv, \
[all …]
Dfill_functor.h31 void operator()(const Device& d, typename TTypes<T>::Flat out,
32 typename TTypes<T>::ConstScalar in);
38 void operator()(const Device& d, typename TTypes<T>::Flat out);
45 typename TTypes<T>::Flat out);
52 void operator()(const Eigen::SyclDevice& d, typename TTypes<T>::Flat out);
59 typename TTypes<string>::Flat out);
65 void operator()(const Device& d, typename TTypes<T>::Flat out);
72 typename TTypes<T>::Flat out);
79 void operator()(const Eigen::SyclDevice& d, typename TTypes<T>::Flat out);
86 typename TTypes<string>::Flat out);
/external/tensorflow/tensorflow/contrib/rnn/kernels/
Dlstm_ops.h32 void operator()(const Device& d, typename TTypes<T>::Flat t) { in operator()
39 void operator()(const Device& d, typename TTypes<T>::UnalignedFlat t) { in operator()
46 void operator()(const Device& d, typename TTypes<T>::ConstFlat src, in operator()
47 typename TTypes<T>::Flat dst) { in operator()
54 void operator()(const Device& d, typename TTypes<T>::UnalignedConstFlat src, in operator()
55 typename TTypes<T>::Flat dst) { in operator()
62 void operator()(const Device& d, typename TTypes<T>::ConstFlat src, in operator()
63 typename TTypes<T>::UnalignedFlat dst) { in operator()
70 void operator()(const Device& d, typename TTypes<T>::ConstFlat a, in operator()
71 typename TTypes<T>::ConstFlat b, typename TTypes<T>::Flat c) { in operator()
[all …]
Dlstm_ops_gpu.cu.cc178 const T cell_clip, bool use_peephole, typename TTypes<T>::ConstMatrix x, in LSTMBlockCellFpropWithCUDA()
179 typename TTypes<T>::ConstMatrix cs_prev, in LSTMBlockCellFpropWithCUDA()
180 typename TTypes<T>::ConstMatrix h_prev, typename TTypes<T>::ConstMatrix w, in LSTMBlockCellFpropWithCUDA()
181 typename TTypes<T>::ConstVec wci, typename TTypes<T>::ConstVec wcf, in LSTMBlockCellFpropWithCUDA()
182 typename TTypes<T>::ConstVec wco, typename TTypes<T>::ConstVec b, in LSTMBlockCellFpropWithCUDA()
183 typename TTypes<T>::Matrix xh, typename TTypes<T>::Matrix i, in LSTMBlockCellFpropWithCUDA()
184 typename TTypes<T>::Matrix cs, typename TTypes<T>::Matrix f, in LSTMBlockCellFpropWithCUDA()
185 typename TTypes<T>::Matrix o, typename TTypes<T>::Matrix ci, in LSTMBlockCellFpropWithCUDA()
186 typename TTypes<T>::Matrix co, typename TTypes<T>::Matrix icfo, in LSTMBlockCellFpropWithCUDA()
187 typename TTypes<T>::Matrix h, int batch_size, int cell_size, in LSTMBlockCellFpropWithCUDA()
[all …]
Dlstm_ops.cc48 typename TTypes<T>::ConstMatrix x, typename TTypes<T>::ConstMatrix cs_prev, in LSTMBlockCellFpropWithEigen()
49 typename TTypes<T>::ConstMatrix h_prev, typename TTypes<T>::ConstMatrix w, in LSTMBlockCellFpropWithEigen()
50 typename TTypes<T>::ConstVec wci, typename TTypes<T>::ConstVec wcf, in LSTMBlockCellFpropWithEigen()
51 typename TTypes<T>::ConstVec wco, typename TTypes<T>::ConstVec b, in LSTMBlockCellFpropWithEigen()
52 typename TTypes<T>::Matrix xh, typename TTypes<T>::Matrix i, in LSTMBlockCellFpropWithEigen()
53 typename TTypes<T>::Matrix cs, typename TTypes<T>::Matrix f, in LSTMBlockCellFpropWithEigen()
54 typename TTypes<T>::Matrix o, typename TTypes<T>::Matrix ci, in LSTMBlockCellFpropWithEigen()
55 typename TTypes<T>::Matrix co, typename TTypes<T>::Matrix icfo, in LSTMBlockCellFpropWithEigen()
56 typename TTypes<T>::Matrix h) { in LSTMBlockCellFpropWithEigen()
62 typename TTypes<T>::ConstMatrix const_xh(xh.data(), xh.dimensions()); in LSTMBlockCellFpropWithEigen()
[all …]
Dgru_ops.h75 OpKernelContext* ctx, const Device& d, typename TTypes<T>::ConstMatrix x, in operator()
76 typename TTypes<T>::ConstMatrix h_prev, in operator()
77 typename TTypes<T>::ConstMatrix w_ru, typename TTypes<T>::ConstMatrix w_c, in operator()
78 typename TTypes<T>::ConstVec b_ru, typename TTypes<T>::ConstVec b_c, in operator()
79 typename TTypes<T>::Matrix r_u_bar, typename TTypes<T>::Matrix r, in operator()
80 typename TTypes<T>::Matrix u, typename TTypes<T>::Matrix c, in operator()
81 typename TTypes<T>::Matrix h, typename TTypes<T>::Matrix x_h_prev, in operator()
82 typename TTypes<T>::Matrix x_h_prevr) { in operator()
88 typename TTypes<T>::ConstMatrix const_x_h_prev(x_h_prev.data(), in operator()
107 typename TTypes<T>::ConstMatrix const_x_h_prevr(x_h_prevr.data(), in operator()
[all …]
Dgru_ops.cc392 typename TTypes<T>::ConstMatrix x, \
393 typename TTypes<T>::ConstMatrix h_prev, \
394 typename TTypes<T>::ConstMatrix w_ru, \
395 typename TTypes<T>::ConstMatrix w_c, typename TTypes<T>::ConstVec b_ru, \
396 typename TTypes<T>::ConstVec b_c, typename TTypes<T>::Matrix r_u_bar, \
397 typename TTypes<T>::Matrix r, typename TTypes<T>::Matrix u, \
398 typename TTypes<T>::Matrix c, typename TTypes<T>::Matrix h, \
399 typename TTypes<T>::Matrix x_h_prev, \
400 typename TTypes<T>::Matrix x_h_prevr); \
422 typename TTypes<T>::ConstMatrix x, typename TTypes<T>::ConstMatrix h, \
[all …]
/external/tensorflow/tensorflow/core/framework/
Dtensor.h242 typename TTypes<T>::Vec vec() { in vec()
247 typename TTypes<T>::Matrix matrix() { in matrix()
252 typename TTypes<T, NDIMS>::Tensor tensor();
260 typename TTypes<T, NDIMS>::Tensor bit_casted_tensor();
270 typename TTypes<T, NDIMS>::Tensor reinterpret_last_dimension();
302 typename TTypes<T>::Flat flat() { in flat()
307 typename TTypes<T>::UnalignedFlat unaligned_flat() { in unaligned_flat()
316 typename TTypes<T, NDIMS>::Tensor flat_inner_dims();
323 typename TTypes<T, NDIMS>::Tensor flat_outer_dims();
332 typename TTypes<T, NDIMS>::Tensor flat_inner_outer_dims(int64 begin);
[all …]

12345678910>>...12