/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) { 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 …]
|
D | sparse_xent_op.h | 31 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 …]
|
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); 50 bool operator()(const Device& d, typename TTypes<float, 4>::ConstTensor grads, 51 typename TTypes<T, 4>::ConstTensor image, [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 | 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() 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 …]
|
D | relu_op.cc | 77 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 …]
|
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 | dense_update_functor.h | 37 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 …]
|
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 | 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 …]
|
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 | gather_nd_op_cpu_impl.h | 45 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 …]
|
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 | batch_norm_op.cc | 180 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 …]
|
D | fill_functor.h | 31 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/ |
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 | 178 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 …]
|
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.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() 107 typename TTypes<T>::ConstMatrix const_x_h_prevr(x_h_prevr.data(), in operator() [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 | 242 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 …]
|