/external/tensorflow/tensorflow/core/kernels/ |
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, 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, 51 bool operator()(const Device& d, typename TTypes<float, 4>::ConstTensor grads, 52 typename TTypes<T, 4>::ConstTensor image, 53 typename TTypes<float, 2>::ConstTensor boxes, 54 typename TTypes<int32, 1>::ConstTensor box_ind, [all …]
|
D | relu_op_functor.h | 33 void operator()(const Device& d, typename TTypes<T>::ConstTensor features, in operator() 48 void operator()(const Device& d, typename TTypes<T>::ConstTensor gradients, in operator() 49 typename TTypes<T>::ConstTensor features, in operator() 66 void operator()(const Device& d, typename TTypes<T>::ConstTensor features, in operator() 81 void operator()(const Device& d, typename TTypes<T>::ConstTensor gradients, in operator() 82 typename TTypes<T>::ConstTensor features, in operator() 101 void operator()(const Device& d, typename TTypes<T>::ConstTensor features, in operator() 116 void operator()(const Device& d, typename TTypes<T>::ConstTensor gradients, in operator() 117 typename TTypes<T>::ConstTensor features, T alpha, in operator() 131 void operator()(const Device& d, typename TTypes<T>::ConstTensor features, 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, 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, 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,
|
D | relu_op.cc | 83 const GPUDevice& d, typename TTypes<T>::ConstTensor features, \ 89 const GPUDevice& d, typename TTypes<T>::ConstTensor gradients, \ 90 typename TTypes<T>::ConstTensor features, \ 96 const GPUDevice& d, typename TTypes<T>::ConstTensor features, \ 102 const GPUDevice& d, typename TTypes<T>::ConstTensor gradients, \ 103 typename TTypes<T>::ConstTensor features, \ 109 const GPUDevice& d, typename TTypes<T>::ConstTensor features, T alpha, \ 115 const GPUDevice& d, typename TTypes<T>::ConstTensor gradients, \ 116 typename TTypes<T>::ConstTensor features, T alpha, \ 122 typename TTypes<T>::ConstTensor features, \ [all …]
|
D | conv_2d.h | 34 typename TTypes<T, Dims, IndexType>::ConstTensor input, in operator() 45 const Device& d, typename TTypes<T, Dims, IndexType>::ConstTensor input, in operator() 71 typename TTypes<T, 4>::ConstTensor input, in operator() 72 typename TTypes<T, 4>::ConstTensor filter, int row_stride, in operator() 85 typename TTypes<Eigen::half, 4>::ConstTensor input, 86 typename TTypes<Eigen::half, 4>::ConstTensor filter, 101 typename TTypes<T, 4>::ConstTensor kernel, 102 typename TTypes<T, 4>::ConstTensor output_backward, 117 typename TTypes<T, 4>::ConstTensor input, 118 typename TTypes<T, 4>::ConstTensor output_backward, [all …]
|
D | betainc_op.h | 29 void operator()(const Device& d, typename TTypes<T, NDIM>::ConstTensor a, in operator() 30 typename TTypes<T, NDIM>::ConstTensor b, in operator() 31 typename TTypes<T, NDIM>::ConstTensor x, in operator() 36 void BCast(const Device& d, typename TTypes<T, NDIM>::ConstTensor a, in BCast() 38 typename TTypes<T, NDIM>::ConstTensor b, in BCast() 40 typename TTypes<T, NDIM>::ConstTensor x, in BCast()
|
D | scale_and_translate_op.h | 62 typename TTypes<int32, 1>::ConstTensor row_starts, 63 typename TTypes<float, 1>::ConstTensor row_weights, 65 typename TTypes<int32, 1>::ConstTensor col_starts, 66 typename TTypes<float, 1>::ConstTensor col_weights, 67 typename TTypes<T, 4>::ConstTensor input_images,
|
D | conv_3d.h | 45 typename TTypes<T, 5>::ConstTensor input, 46 typename TTypes<T, 5>::ConstTensor filter, int stride_planes, 58 typename TTypes<T, 5>::ConstTensor filter, 59 typename TTypes<T, 5>::ConstTensor output_backward, 75 typename TTypes<T, 5>::ConstTensor input, 76 typename TTypes<T, 5>::ConstTensor output_backward,
|
D | matrix_diag_op.h | 33 MatrixDiagPartGenerator(typename TTypes<T, 3>::ConstTensor input) in MatrixDiagPartGenerator() 44 typename TTypes<T, 3>::ConstTensor input_; 51 MatrixDiagGenerator(typename TTypes<T, 2>::ConstTensor input) in MatrixDiagGenerator() 63 typename TTypes<T, 2>::ConstTensor input_; 73 const Device& d, typename TTypes<T, 3>::ConstTensor input, in Compute() 83 const Device& d, typename TTypes<T, 2>::ConstTensor input, in Compute()
|
D | searchsorted_op.h | 33 const typename TTypes<T, 1>::ConstTensor& sorted_inputs, 34 const typename TTypes<T, 1>::ConstTensor& values, 44 const typename TTypes<T, 1>::ConstTensor& sorted_inputs, 45 const typename TTypes<T, 1>::ConstTensor& values,
|
D | scatter_nd_op_cpu_impl.h | 96 typename TTypes<Index, 2>::ConstTensor Tindices, 97 typename TTypes<T, 2>::ConstTensor Tupdates, 147 typename TTypes<Index, 2>::ConstTensor Tindices, \ 148 typename TTypes<T, 2>::ConstTensor Tupdates, \ 180 typename TTypes<Index, 2>::ConstTensor Tindices, 181 typename TTypes<T, 2>::ConstTensor Tupdates, 231 typename TTypes<Index, 2>::ConstTensor Tindices, \ 232 typename TTypes<T, 2>::ConstTensor Tupdates, \
|
D | betainc_op.cc | 131 const GPUDevice& d, typename TTypes<T, NDIM>::ConstTensor a, \ 132 typename TTypes<T, NDIM>::ConstTensor b, \ 133 typename TTypes<T, NDIM>::ConstTensor x, \ 137 const GPUDevice& d, typename TTypes<T, NDIM>::ConstTensor a, \ 139 typename TTypes<T, NDIM>::ConstTensor b, \ 141 typename TTypes<T, NDIM>::ConstTensor x, \
|
D | split_lib.h | 29 typename TTypes<T, 2>::ConstTensor input, 37 typename TTypes<T, NDims>::ConstTensor input, 46 typename TTypes<T, NDims>::ConstTensor input, 56 typename TTypes<T, NDims>::ConstTensor input,
|
D | cross_op.cc | 71 typename TTypes<Type, 2>::ConstTensor in0_data = in Compute() 73 typename TTypes<Type, 2>::ConstTensor in1_data = in Compute() 97 const GPUDevice& d, TTypes<type, 2>::ConstTensor in0_data, \ 98 TTypes<type, 2>::ConstTensor in1_data, \
|
D | scale_and_translate_op.cc | 334 typename TTypes<T, 4>::ConstTensor image_data(input.tensor<T, 4>()); in Compute() 357 typename TTypes<int32, 1>::ConstTensor row_starts( in Compute() 359 typename TTypes<float, 1>::ConstTensor row_weights( in Compute() 362 typename TTypes<int32, 1>::ConstTensor col_starts( in Compute() 364 typename TTypes<float, 1>::ConstTensor col_weights( in Compute() 436 TTypes<float, 4>::ConstTensor input_grad = input.tensor<float, 4>(); in Compute() 462 typename TTypes<int32, 1>::ConstTensor row_starts = in Compute() 464 typename TTypes<float, 1>::ConstTensor row_weights( in Compute() 467 typename TTypes<int32, 1>::ConstTensor col_starts( in Compute() 469 typename TTypes<float, 1>::ConstTensor col_weights( in Compute() [all …]
|
D | softsign_op.h | 34 void operator()(const Device& d, typename TTypes<T>::ConstTensor features, in operator() 49 void operator()(const Device& d, typename TTypes<T>::ConstTensor gradients, in operator() 50 typename TTypes<T>::ConstTensor features, in operator()
|
D | eigen_benchmark.h | 32 using Input = TTypes<float, 4>::ConstTensor; 33 using Filter = TTypes<float, 4>::ConstTensor; 79 using OutputBackward = TTypes<float, 4>::ConstTensor; in SpatialConvolutionBackwardInput() 120 using OutputBackward = TTypes<float, 4>::ConstTensor; in SpatialConvolutionBackwardKernel() 167 using Input = TTypes<float, 5>::ConstTensor; 168 using Filter = TTypes<float, 5>::ConstTensor; 221 using OutputBackward = TTypes<float, 5>::ConstTensor; in CuboidConvolutionBackwardInput() 258 using OutputBackward = TTypes<float, 5>::ConstTensor; in CuboidConvolutionBackwardKernel()
|
D | dilation_ops_gpu.cu.cc | 181 void operator()(const GPUDevice& d, typename TTypes<T, 4>::ConstTensor input, in operator ()() 182 typename TTypes<T, 3>::ConstTensor filter, int stride_rows, in operator ()() 210 void operator()(const GPUDevice& d, typename TTypes<T, 4>::ConstTensor input, in operator ()() 211 typename TTypes<T, 3>::ConstTensor filter, in operator ()() 212 typename TTypes<T, 4>::ConstTensor out_backprop, in operator ()() 252 void operator()(const GPUDevice& d, typename TTypes<T, 4>::ConstTensor input, in operator ()() 253 typename TTypes<T, 3>::ConstTensor filter, in operator ()() 254 typename TTypes<T, 4>::ConstTensor out_backprop, in operator ()()
|
D | crop_and_resize_op_gpu.cu.cc | 350 typename TTypes<T, 4>::ConstTensor image, in operator ()() 351 typename TTypes<float, 2>::ConstTensor boxes, in operator ()() 352 typename TTypes<int32, 1>::ConstTensor box_ind, in operator ()() 388 typename TTypes<float, 4>::ConstTensor grads, in operator ()() 389 typename TTypes<float, 2>::ConstTensor boxes, in operator ()() 390 typename TTypes<int32, 1>::ConstTensor box_ind, in operator ()() 438 typename TTypes<float, 4>::ConstTensor grads, in operator ()() 439 typename TTypes<T, 4>::ConstTensor image, in operator ()() 440 typename TTypes<float, 2>::ConstTensor boxes, in operator ()() 441 typename TTypes<int32, 1>::ConstTensor box_ind, in operator ()()
|
D | softplus_op.h | 34 void operator()(const Device& d, typename TTypes<T>::ConstTensor features, in operator() 65 void operator()(const Device& d, typename TTypes<T>::ConstTensor gradients, in operator() 66 typename TTypes<T>::ConstTensor features, in operator()
|
D | searchsorted_op_gpu.cu.cc | 63 const typename TTypes<T, 1>::ConstTensor& sorted_inputs, in Compute() 64 const typename TTypes<T, 1>::ConstTensor& values, in Compute() 83 const typename TTypes<T, 1>::ConstTensor& sorted_inputs, in Compute() 84 const typename TTypes<T, 1>::ConstTensor& values, in Compute()
|
D | crop_and_resize_op.cc | 83 OpKernelContext* context, typename TTypes<int32, 1>::ConstTensor box_index, 89 OpKernelContext* context, typename TTypes<int32, 1>::ConstTensor box_index, in RunIfBoxIndexIsValid() 204 typename TTypes<T, 4>::ConstTensor image, in operator ()() 205 typename TTypes<float, 2>::ConstTensor boxes, in operator ()() 206 typename TTypes<int32, 1>::ConstTensor box_index, in operator ()() 429 typename TTypes<float, 4>::ConstTensor grads, in operator ()() 430 typename TTypes<float, 2>::ConstTensor boxes, in operator ()() 431 typename TTypes<int32, 1>::ConstTensor box_index, in operator ()() 604 typename TTypes<float, 4>::ConstTensor grads, in operator ()() 605 typename TTypes<T, 4>::ConstTensor image, in operator ()() [all …]
|
D | gather_functor.h | 40 typename TTypes<T, 3>::ConstTensor params, in HandleCopies() 115 typename TTypes<T, 3>::ConstTensor params, in operator() 152 typename TTypes<T, 3>::ConstTensor params, 160 typename TTypes<T, 3>::ConstTensor params, 170 typename TTypes<Variant, 3>::ConstTensor params,
|
/external/tensorflow/tensorflow/core/framework/ |
D | tensor.h | 465 typename TTypes<T, NDIMS>::ConstTensor tensor() const; 473 typename TTypes<T, NDIMS>::ConstTensor bit_casted_tensor() const; 483 typename TTypes<T, NDIMS>::ConstTensor reinterpret_last_dimension() const; 496 typename TTypes<T, NDIMS>::ConstTensor shaped( 505 typename TTypes<T, NDIMS>::ConstTensor bit_casted_shaped( 516 typename TTypes<T, NDIMS>::ConstTensor flat_inner_dims() const; 519 typename TTypes<T, NDIMS>::ConstTensor flat_outer_dims() const; 522 typename TTypes<T, NDIMS>::ConstTensor flat_inner_outer_dims( 723 typename TTypes<T, NDIMS>::ConstTensor Tensor::tensor() const { in tensor() 725 return typename TTypes<T, NDIMS>::ConstTensor(base<const T>(), in tensor() [all …]
|
/external/tensorflow/tensorflow/contrib/seq2seq/kernels/ |
D | beam_search_ops.cc | 83 typename TTypes<T, 3>::ConstTensor step_ids_t(step_ids.tensor<T, 3>()); in Compute() 84 typename TTypes<T, 3>::ConstTensor parent_ids_t(parent_ids.tensor<T, 3>()); in Compute() 108 TTypes<int32, 3>::ConstTensor step_ids, in operator ()() 109 TTypes<int32, 3>::ConstTensor parent_ids, in operator ()() 174 typename TTypes<T, 3>::ConstTensor step_ids, \ 175 typename TTypes<T, 3>::ConstTensor parent_ids, \
|