Home
last modified time | relevance | path

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

12345678910>>...14

/external/tensorflow/tensorflow/core/kernels/
Dtraining_ops.h33 void operator()(const Device& d, typename TTypes<T>::Flat var,
34 typename TTypes<T>::ConstScalar alpha,
35 typename TTypes<T>::ConstFlat delta);
40 void operator()(const Device& d, typename TTypes<T>::Flat var,
41 typename TTypes<T>::Flat accum,
42 typename TTypes<T>::Flat accum_update,
43 typename TTypes<T>::ConstScalar lr,
44 typename TTypes<T>::ConstScalar rho,
45 typename TTypes<T>::ConstScalar epsilon,
46 typename TTypes<T>::ConstFlat grad);
[all …]
Daggregate_ops.h30 void operator()(const Device& d, typename TTypes<T>::Flat out,
31 typename TTypes<T>::ConstFlat in1,
32 typename TTypes<T>::ConstFlat in2);
37 static void Compute(const Device& d, typename TTypes<T>::Flat out, in Compute()
38 typename TTypes<T>::ConstFlat in1, in Compute()
39 typename TTypes<T>::ConstFlat in2) { in Compute()
46 void operator()(const Device& d, typename TTypes<T>::Flat out,
47 typename TTypes<T>::ConstFlat in1,
48 typename TTypes<T>::ConstFlat in2,
49 typename TTypes<T>::ConstFlat in3);
[all …]
Daggregate_ops_gpu.cu.cc35 void operator()(const GPUDevice& d, typename TTypes<T>::Flat out, in operator ()()
36 typename TTypes<T>::ConstFlat in1, in operator ()()
37 typename TTypes<T>::ConstFlat in2) { in operator ()()
44 void operator()(const GPUDevice& d, typename TTypes<T>::Flat out, 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 ()()
54 void operator()(const GPUDevice& d, typename TTypes<T>::Flat out, in operator ()()
55 typename TTypes<T>::ConstFlat in1, in operator ()()
56 typename TTypes<T>::ConstFlat in2, in operator ()()
[all …]
Daggregate_ops_cpu.h34 void operator()(const CPUDevice& d, typename TTypes<T>::Flat out,
35 typename TTypes<T>::ConstFlat in1,
36 typename TTypes<T>::ConstFlat in2) {
42 void operator()(const CPUDevice& d, typename TTypes<T>::Flat out,
43 typename TTypes<T>::ConstFlat in1,
44 typename TTypes<T>::ConstFlat in2,
45 typename TTypes<T>::ConstFlat in3) {
51 void operator()(const CPUDevice& d, typename TTypes<T>::Flat out,
52 typename TTypes<T>::ConstFlat in1,
53 typename TTypes<T>::ConstFlat in2,
[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 …]
Drelu_op_functor.h33 void operator()(const Device& d, typename TTypes<T>::ConstTensor features, in operator()
34 typename TTypes<T>::Tensor activations) { in operator()
49 void operator()(const Device& d, typename TTypes<T>::ConstTensor gradients, in operator()
50 typename TTypes<T>::ConstTensor features, in operator()
51 typename TTypes<T>::Tensor backprops) { in operator()
67 void operator()(const Device& d, typename TTypes<T>::ConstTensor features, in operator()
68 typename TTypes<T>::Tensor activations) { in operator()
83 void operator()(const Device& d, typename TTypes<T>::ConstTensor gradients, in operator()
84 typename TTypes<T>::ConstTensor features, in operator()
85 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 …]
Dtraining_ops_gpu.cu.cc338 void operator()(const GPUDevice& d, typename TTypes<T>::Flat var, in operator ()()
339 typename TTypes<T>::ConstScalar lr, in operator ()()
340 typename TTypes<T>::ConstFlat grad) { in operator ()()
341 Eigen::array<typename TTypes<T>::Tensor::Index, 1> bcast; in operator ()()
460 void operator()(const GPUDevice& d, typename TTypes<T>::Flat var, in operator ()()
461 typename TTypes<T>::Flat accum, in operator ()()
462 typename TTypes<T>::ConstScalar lr, in operator ()()
463 typename TTypes<T>::ConstFlat grad, bool update_slots) { in operator ()()
471 Eigen::array<typename TTypes<T>::Tensor::Index, 1> bcast; in operator ()()
481 void operator()(const GPUDevice& d, typename TTypes<T>::Flat var, in operator ()()
[all …]
Dquantize_and_dequantize_op_gpu.cu.cc32 void operator()(const GPUDevice& d, typename TTypes<T>::ConstVec input, in operator ()()
36 typename TTypes<T>::Vec output) { in operator ()()
45 void operator()(const GPUDevice& d, typename TTypes<T, 3>::ConstTensor input, in operator ()()
49 typename TTypes<T, 3>::Tensor output) { in operator ()()
58 void operator()(const GPUDevice& d, typename TTypes<T>::ConstFlat gradient, in operator ()()
59 typename TTypes<T>::ConstFlat input, in operator ()()
60 typename TTypes<T>::ConstScalar input_min_tensor, in operator ()()
61 typename TTypes<T>::ConstScalar input_max_tensor, in operator ()()
62 typename TTypes<T>::Flat input_backprop, in operator ()()
63 typename TTypes<T>::Scalar input_min_backprop, in operator ()()
[all …]
Dquantize_and_dequantize_op.h47 void operator()(const Device& d, typename TTypes<T>::ConstVec input,
51 typename TTypes<T>::Vec output);
56 void operator()(const Device& d, typename TTypes<T, 3>::ConstTensor input,
60 typename TTypes<T, 3>::Tensor output);
65 void operator()(const Device& d, typename TTypes<T>::ConstFlat gradient,
66 typename TTypes<T>::ConstFlat input,
67 typename TTypes<T>::ConstScalar input_min,
68 typename TTypes<T>::ConstScalar input_max,
69 typename TTypes<T>::Flat input_backprop,
70 typename TTypes<T>::Scalar input_min_backprop,
[all …]
Drelu_op.cc81 const GPUDevice& d, typename TTypes<T>::ConstTensor features, \
82 typename TTypes<T>::Tensor activations); \
87 const GPUDevice& d, typename TTypes<T>::ConstTensor gradients, \
88 typename TTypes<T>::ConstTensor features, \
89 typename TTypes<T>::Tensor backprops); \
94 const GPUDevice& d, typename TTypes<T>::ConstTensor features, \
95 typename TTypes<T>::Tensor activations); \
100 const GPUDevice& d, typename TTypes<T>::ConstTensor gradients, \
101 typename TTypes<T>::ConstTensor features, \
102 typename TTypes<T>::Tensor backprops); \
[all …]
Dcwise_op_gpu_select.cu.cc29 typename TTypes<T, NDIMS>::Tensor output_tensor, in operator ()()
30 typename TTypes<bool, NDIMS>::ConstTensor cond_tensor, in operator ()()
31 typename TTypes<T, NDIMS>::ConstTensor then_tensor, in operator ()()
32 typename TTypes<T, NDIMS>::ConstTensor else_tensor, in operator ()()
44 void operator()(const GPUDevice& d, typename TTypes<T>::Flat out, in operator ()()
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 ()()
55 void operator()(const GPUDevice& d, typename TTypes<T>::Flat out, in operator ()()
56 typename TTypes<bool>::ConstScalar cond, 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 …]
Dconv_2d.h49 void operator()(const Device& d, typename TTypes<T, 4>::Tensor output, in operator()
50 typename TTypes<T, 4>::ConstTensor input, in operator()
51 typename TTypes<T, 4>::ConstTensor filter, int row_stride, in operator()
58 void operator()(const Device& d, typename TTypes<T, 4>::Tensor output, in operator()
59 typename TTypes<T, 4>::ConstTensor input, in operator()
60 typename TTypes<T, 4>::ConstTensor filter, int row_stride, in operator()
75 typename TTypes<Eigen::half, 4>::Tensor output,
76 typename TTypes<Eigen::half, 4>::ConstTensor input,
77 typename TTypes<Eigen::half, 4>::ConstTensor filter,
88 typename TTypes<Eigen::half, 4>::Tensor output,
[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 …]
Dparameterized_truncated_normal_op.h40 typename TTypes<T>::ConstFlat means,
41 typename TTypes<T>::ConstFlat stddevs,
42 typename TTypes<T>::ConstFlat minvals,
43 typename TTypes<T>::ConstFlat maxvals,
45 typename TTypes<T>::Flat output);
55 typename TTypes<T>::ConstFlat means,
56 typename TTypes<T>::ConstFlat stddevs,
57 typename TTypes<T>::ConstFlat minvals,
58 typename TTypes<T>::ConstFlat maxvals,
60 typename TTypes<T>::Flat output);
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_clip_gpu.cu.cc67 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 ()()
70 typename TTypes<T>::Flat &out_flat) const { 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 ()()
86 typename TTypes<T>::Flat &out_flat) const { 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 ()()
[all …]
/external/tensorflow/tensorflow/core/kernels/rnn/
Dlstm_ops.h42 void operator()(const Device& d, typename TTypes<T>::Flat t) { in operator()
49 void operator()(const Device& d, typename TTypes<T>::UnalignedFlat t) { in operator()
56 void operator()(const Device& d, typename TTypes<T>::ConstFlat src, in operator()
57 typename TTypes<T>::Flat dst) { in operator()
64 void operator()(const Device& d, typename TTypes<T>::UnalignedConstFlat src, in operator()
65 typename TTypes<T>::Flat dst) { in operator()
72 void operator()(const Device& d, typename TTypes<T>::ConstFlat src, in operator()
73 typename TTypes<T>::UnalignedFlat dst) { in operator()
80 void operator()(const Device& d, typename TTypes<T>::ConstFlat a, in operator()
81 typename TTypes<T>::ConstFlat b, typename TTypes<T>::Flat c) { in operator()
[all …]
Dlstm_ops_gpu.cu.cc62 void operator()(const GPUDevice& d, typename TTypes<T>::Flat t) { in operator ()()
69 void operator()(const GPUDevice& d, typename TTypes<T>::UnalignedFlat t) { in operator ()()
229 const float cell_clip, bool use_peephole, typename TTypes<T>::ConstMatrix x, in LSTMBlockCellFpropWithCUDA()
230 typename TTypes<T>::ConstMatrix cs_prev, in LSTMBlockCellFpropWithCUDA()
231 typename TTypes<T>::ConstMatrix h_prev, typename TTypes<T>::ConstMatrix w, in LSTMBlockCellFpropWithCUDA()
232 typename TTypes<T>::ConstVec wci, typename TTypes<T>::ConstVec wcf, in LSTMBlockCellFpropWithCUDA()
233 typename TTypes<T>::ConstVec wco, typename TTypes<T>::ConstVec b, in LSTMBlockCellFpropWithCUDA()
234 typename TTypes<T>::Matrix xh, typename TTypes<T>::Matrix i, in LSTMBlockCellFpropWithCUDA()
235 typename TTypes<T>::Matrix cs, typename TTypes<T>::Matrix f, in LSTMBlockCellFpropWithCUDA()
236 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 gates, in LSTMBlockCellFpropWithEigen()
56 typename TTypes<T>::Matrix h) { in LSTMBlockCellFpropWithEigen()
62 typename TTypes<T>::ConstMatrix const_xh(xh.data(), xh.dimensions()); in LSTMBlockCellFpropWithEigen()
[all …]
/external/tensorflow/tensorflow/core/kernels/sparse/
Dkernels.h39 Status operator()(OpKernelContext* c, TTypes<int64>::ConstMatrix indices,
40 TTypes<int32>::Vec nnz_per_batch);
59 void operator()(const Device& d, TTypes<int64>::ConstVec host_dense_shape,
60 TTypes<int64>::ConstMatrix indices,
61 TTypes<int32>::Vec coo_row_ind,
62 TTypes<int32>::Vec coo_col_ind);
86 TTypes<int64>::ConstVec host_dense_shape,
87 TTypes<int32>::ConstVec host_batch_ptrs,
88 TTypes<int32>::Vec coo_row_ind,
89 TTypes<int32>::ConstVec coo_col_ind,
[all …]
/external/tensorflow/tensorflow/core/kernels/image/
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);
42 typename TTypes<float, 4>::ConstTensor grads,
43 typename TTypes<float, 2>::ConstTensor boxes,
44 typename TTypes<int32, 1>::ConstTensor box_ind,
45 typename TTypes<T, 4>::Tensor grads_image,
52 bool operator()(const Device& d, typename TTypes<float, 4>::ConstTensor grads,
53 typename TTypes<T, 4>::ConstTensor image,
[all …]
/external/tensorflow/tensorflow/core/framework/
Dtensor.h400 typename TTypes<T>::Vec vec() { in vec()
405 typename TTypes<T>::Matrix matrix() { in matrix()
410 typename TTypes<T, NDIMS>::Tensor tensor();
418 typename TTypes<T, NDIMS>::Tensor bit_casted_tensor();
428 typename TTypes<T, NDIMS>::Tensor reinterpret_last_dimension();
460 typename TTypes<T>::Flat flat() { in flat()
465 typename TTypes<T>::UnalignedFlat unaligned_flat() { in unaligned_flat()
474 typename TTypes<T, NDIMS>::Tensor flat_inner_dims();
481 typename TTypes<T, NDIMS>::Tensor flat_outer_dims();
490 typename TTypes<T, NDIMS>::Tensor flat_inner_outer_dims(int64 begin);
[all …]

12345678910>>...14