Home
last modified time | relevance | path

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

12345678910>>...13

/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, bool update_slots) { in operator ()()
49 Eigen::array<typename TTypes<T>::Tensor::Index, 1> bcast; in operator ()()
58 void operator()(const GPUDevice& d, typename TTypes<T>::Flat var, in operator ()()
[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 …]
Dsparse_xent_op.h32 typename TTypes<const T, 1>::Tensor32Bit To32BitConst( in To32BitConst()
33 typename TTypes<T>::Vec in) { in To32BitConst()
34 return To32Bit(typename TTypes<T>::ConstVec(in.data(), in.dimensions())); in To32BitConst()
38 typename TTypes<const T, 2>::Tensor32Bit To32BitConst( in To32BitConst()
39 typename TTypes<T>::Matrix in) { in To32BitConst()
40 return To32Bit(typename TTypes<T>::ConstMatrix(in.data(), in.dimensions())); in To32BitConst()
60 typename TTypes<const T, 2>::Tensor32Bit logits, in SparseXentLossGenerator()
61 typename TTypes<const T, 1>::Tensor32Bit sum_exp_logits, in SparseXentLossGenerator()
62 typename TTypes<const Index, 1>::Tensor32Bit labels, in SparseXentLossGenerator()
83 typename TTypes<const T, 2>::Tensor32Bit logits_;
[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,
29 typename TTypes<T>::Flat &out_flat) const;
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,
38 typename TTypes<T>::Flat &out_flat) const;
44 void operator()(const Device &d, typename TTypes<T>::ConstFlat &in0_flat,
45 typename TTypes<T>::ConstFlat &in1_flat,
[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,
51 bool operator()(const Device& d, typename TTypes<float, 4>::ConstTensor grads,
52 typename TTypes<T, 4>::ConstTensor image,
[all …]
Drelu_op.cc83 const GPUDevice& d, typename TTypes<T>::ConstTensor features, \
84 typename TTypes<T>::Tensor activations); \
89 const GPUDevice& d, typename TTypes<T>::ConstTensor gradients, \
90 typename TTypes<T>::ConstTensor features, \
91 typename TTypes<T>::Tensor backprops); \
96 const GPUDevice& d, typename TTypes<T>::ConstTensor features, \
97 typename TTypes<T>::Tensor activations); \
102 const GPUDevice& d, typename TTypes<T>::ConstTensor gradients, \
103 typename TTypes<T>::ConstTensor features, \
104 typename TTypes<T>::Tensor backprops); \
[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()
70 void operator()(const Device& d, typename TTypes<T, 4>::Tensor output, in operator()
71 typename TTypes<T, 4>::ConstTensor input, in operator()
72 typename TTypes<T, 4>::ConstTensor filter, int row_stride, in operator()
84 typename TTypes<Eigen::half, 4>::Tensor output,
85 typename TTypes<Eigen::half, 4>::ConstTensor input,
86 typename TTypes<Eigen::half, 4>::ConstTensor filter,
[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 …]
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 ()()
62 typename TTypes<T>::Flat &out_flat) const { 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 ()()
78 typename TTypes<T>::Flat &out_flat) const { 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 ()()
[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 …]
Ddense_update_functor.h40 void operator()(const Device& d, typename TTypes<T>::Flat params,
41 typename TTypes<T>::ConstFlat update);
46 void operator()(const CPUDevice& d, typename TTypes<T>::Flat params,
47 typename TTypes<T>::ConstFlat update) {
54 void operator()(const CPUDevice& d, typename TTypes<T>::Flat params,
55 typename TTypes<T>::ConstFlat update) {
62 void operator()(const CPUDevice& d, typename TTypes<T>::Flat params,
63 typename TTypes<T>::ConstFlat update) {
71 void operator()(const SYCLDevice& d, typename TTypes<T>::Flat params,
72 typename TTypes<T>::ConstFlat update) {
[all …]
Dsparse_utils_test.cc31 using tensorflow::TTypes;
43 TTypes<int32>::ConstMatrix indices_mat(data, 8, 2); in TEST()
54 TTypes<int32>::ConstMatrix indices_mat(data, 15, 2); in TEST()
65 TTypes<int64>::ConstMatrix indices_mat(data, 1, 2); in TEST()
74 TTypes<uint32>::ConstMatrix indices_mat(data, 2, 2); in TEST()
83 TTypes<uint16>::ConstMatrix indices_mat(data, 4, 2); in TEST()
93 TTypes<uint64>::ConstMatrix indices_mat(data, 4, 2); in TEST()
145 TTypes<int32>::ConstMatrix indices_mat(data, 8, 2); in TEST()
154 TTypes<int64>::ConstMatrix indices_mat(data, 8, 2); in TEST()
163 TTypes<int32>::ConstMatrix indices_mat(data, 6, 2); in TEST()
[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 …]
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()
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 …]
/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.cc63 void operator()(const GPUDevice& d, typename TTypes<T>::Flat t) { in operator ()()
70 void operator()(const GPUDevice& d, typename TTypes<T>::UnalignedFlat t) { in operator ()()
224 const float cell_clip, bool use_peephole, typename TTypes<T>::ConstMatrix x, in LSTMBlockCellFpropWithCUDA()
225 typename TTypes<T>::ConstMatrix cs_prev, in LSTMBlockCellFpropWithCUDA()
226 typename TTypes<T>::ConstMatrix h_prev, typename TTypes<T>::ConstMatrix w, in LSTMBlockCellFpropWithCUDA()
227 typename TTypes<T>::ConstVec wci, typename TTypes<T>::ConstVec wcf, in LSTMBlockCellFpropWithCUDA()
228 typename TTypes<T>::ConstVec wco, typename TTypes<T>::ConstVec b, in LSTMBlockCellFpropWithCUDA()
229 typename TTypes<T>::Matrix xh, typename TTypes<T>::Matrix i, in LSTMBlockCellFpropWithCUDA()
230 typename TTypes<T>::Matrix cs, typename TTypes<T>::Matrix f, in LSTMBlockCellFpropWithCUDA()
231 typename TTypes<T>::Matrix o, typename TTypes<T>::Matrix ci, in LSTMBlockCellFpropWithCUDA()
[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()
109 typename TTypes<T>::ConstMatrix const_x_h_prevr(x_h_prevr.data(), in operator()
[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.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.h337 typename TTypes<T>::Vec vec() { in vec()
342 typename TTypes<T>::Matrix matrix() { in matrix()
347 typename TTypes<T, NDIMS>::Tensor tensor();
355 typename TTypes<T, NDIMS>::Tensor bit_casted_tensor();
365 typename TTypes<T, NDIMS>::Tensor reinterpret_last_dimension();
397 typename TTypes<T>::Flat flat() { in flat()
402 typename TTypes<T>::UnalignedFlat unaligned_flat() { in unaligned_flat()
411 typename TTypes<T, NDIMS>::Tensor flat_inner_dims();
418 typename TTypes<T, NDIMS>::Tensor flat_outer_dims();
427 typename TTypes<T, NDIMS>::Tensor flat_inner_outer_dims(int64 begin);
[all …]

12345678910>>...13