/external/tensorflow/tensorflow/core/kernels/ |
D | training_ops.h | 32 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 …]
|
D | aggregate_ops.h | 29 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 …]
|
D | aggregate_ops_cpu.h | 37 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 …]
|
D | aggregate_ops_gpu.cu.cc | 34 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 …]
|
D | training_ops_gpu.cu.cc | 30 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 …]
|
D | relu_op_functor.h | 33 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 …]
|
D | sparse_xent_op.h | 32 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 …]
|
D | cwise_op_clip.h | 26 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 …]
|
D | crop_and_resize_op.h | 31 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 …]
|
D | relu_op.cc | 83 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 …]
|
D | conv_2d.h | 34 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 …]
|
D | batch_norm_op.h | 28 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 …]
|
D | cwise_op_clip_gpu.cu.cc | 59 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 …]
|
D | dilation_ops.h | 29 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 …]
|
D | dense_update_functor.h | 40 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 …]
|
D | sparse_utils_test.cc | 31 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 …]
|
D | cwise_op_gpu_select.cu.cc | 28 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 …]
|
D | one_hot_op.h | 34 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()
|
D | fake_quant_ops.cc | 39 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/ |
D | lstm_ops.h | 32 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 …]
|
D | lstm_ops_gpu.cu.cc | 63 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 …]
|
D | gru_ops.h | 75 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 …]
|
D | lstm_ops.cc | 48 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 …]
|
D | gru_ops.cc | 392 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/ |
D | tensor.h | 337 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 …]
|