| /external/tensorflow/tensorflow/core/kernels/image/ |
| 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,
42 typename TTypes<float, 4>::ConstTensor grads,
43 typename TTypes<float, 2>::ConstTensor boxes,
44 typename TTypes<int32, 1>::ConstTensor box_ind,
52 bool operator()(const Device& d, typename TTypes<float, 4>::ConstTensor grads,
53 typename TTypes<T, 4>::ConstTensor image,
54 typename TTypes<float, 2>::ConstTensor boxes,
55 typename TTypes<int32, 1>::ConstTensor box_ind,
[all …]
|
| 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 | scale_and_translate_op.cc | 335 typename TTypes<T, 4>::ConstTensor image_data(input.tensor<T, 4>()); in Compute()
358 typename TTypes<int32, 1>::ConstTensor row_starts( in Compute()
360 typename TTypes<float, 1>::ConstTensor row_weights( in Compute()
363 typename TTypes<int32, 1>::ConstTensor col_starts( in Compute()
365 typename TTypes<float, 1>::ConstTensor col_weights( in Compute()
437 TTypes<float, 4>::ConstTensor input_grad = input.tensor<float, 4>(); in Compute()
463 typename TTypes<int32, 1>::ConstTensor row_starts = in Compute()
465 typename TTypes<float, 1>::ConstTensor row_weights( in Compute()
468 typename TTypes<int32, 1>::ConstTensor col_starts( in Compute()
470 typename TTypes<float, 1>::ConstTensor col_weights( in Compute()
[all …]
|
| D | crop_and_resize_op_gpu.cu.cc | 352 typename TTypes<T, 4>::ConstTensor image, in operator ()()
353 typename TTypes<float, 2>::ConstTensor boxes, in operator ()()
354 typename TTypes<int32, 1>::ConstTensor box_ind, in operator ()()
390 typename TTypes<float, 4>::ConstTensor grads, in operator ()()
391 typename TTypes<float, 2>::ConstTensor boxes, in operator ()()
392 typename TTypes<int32, 1>::ConstTensor box_ind, in operator ()()
441 typename TTypes<float, 4>::ConstTensor grads, in operator ()()
442 typename TTypes<T, 4>::ConstTensor image, in operator ()()
443 typename TTypes<float, 2>::ConstTensor boxes, in operator ()()
444 typename TTypes<int32, 1>::ConstTensor box_ind, in operator ()()
|
| D | colorspace_op.cc | 63 typename TTypes<T, 2>::ConstTensor input_data = input.flat_inner_dims<T>(); in Compute()
100 typename TTypes<T, 2>::ConstTensor input_data = input.flat_inner_dims<T>(); in Compute()
128 const GPUDevice& d, TTypes<T, 2>::ConstTensor input_data, \
133 const GPUDevice& d, TTypes<T, 2>::ConstTensor input_data, \
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | relu_op_functor.h | 33 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()
67 void operator()(const Device& d, typename TTypes<T>::ConstTensor features, in operator()
83 void operator()(const Device& d, typename TTypes<T>::ConstTensor gradients, in operator()
84 typename TTypes<T>::ConstTensor features, in operator()
109 typename TTypes<T>::ConstTensor features;
130 void operator()(const Device& d, typename TTypes<T>::ConstTensor gradients, in operator()
131 typename TTypes<T>::ConstTensor features, T alpha, in operator()
145 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 | 81 const GPUDevice& d, typename TTypes<T>::ConstTensor features, \
87 const GPUDevice& d, typename TTypes<T>::ConstTensor gradients, \
88 typename TTypes<T>::ConstTensor features, \
94 const GPUDevice& d, typename TTypes<T>::ConstTensor features, \
100 const GPUDevice& d, typename TTypes<T>::ConstTensor gradients, \
101 typename TTypes<T>::ConstTensor features, \
111 const GPUDevice& d, typename TTypes<T>::ConstTensor gradients, \
112 typename TTypes<T>::ConstTensor features, T alpha, \
118 typename TTypes<T>::ConstTensor features, \
124 const GPUDevice& d, typename TTypes<T>::ConstTensor gradients, \
[all …]
|
| D | conv_2d.h | 50 typename TTypes<T, 4>::ConstTensor input, in operator()
51 typename TTypes<T, 4>::ConstTensor filter, int row_stride, in operator()
59 typename TTypes<T, 4>::ConstTensor input, in operator()
60 typename TTypes<T, 4>::ConstTensor filter, int row_stride, in operator()
76 typename TTypes<Eigen::half, 4>::ConstTensor input,
77 typename TTypes<Eigen::half, 4>::ConstTensor filter,
89 typename TTypes<Eigen::half, 4>::ConstTensor input,
90 typename TTypes<Eigen::half, 4>::ConstTensor filter,
108 typename TTypes<T, 4>::ConstTensor filter,
109 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 | 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 | 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 | bincount_op.h | 32 const typename TTypes<Tidx, 1>::ConstTensor& arr,
33 const typename TTypes<T, 1>::ConstTensor& weights,
41 const typename TTypes<Tidx, 2>::ConstTensor& in,
42 const typename TTypes<T, 2>::ConstTensor& weights,
|
| 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 | betainc_op.cc | 132 const GPUDevice& d, typename TTypes<T, NDIM>::ConstTensor a, \
133 typename TTypes<T, NDIM>::ConstTensor b, \
134 typename TTypes<T, NDIM>::ConstTensor x, \
138 const GPUDevice& d, typename TTypes<T, NDIM>::ConstTensor a, \
140 typename TTypes<T, NDIM>::ConstTensor b, \
142 typename TTypes<T, NDIM>::ConstTensor x, \
|
| 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 | softplus_op.h | 37 void operator()(const Device& d, typename TTypes<T>::ConstTensor features, in operator()
68 void operator()(const Device& d, typename TTypes<T>::ConstTensor gradients, in operator()
69 typename TTypes<T>::ConstTensor features, in operator()
|
| 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,
|
| D | eigen_benchmark.h | 32 using Input = TTypes<float, 4>::ConstTensor;
33 using Filter = TTypes<float, 4>::ConstTensor;
78 using OutputBackward = TTypes<float, 4>::ConstTensor; in SpatialConvolutionBackwardInput()
117 using OutputBackward = TTypes<float, 4>::ConstTensor; in SpatialConvolutionBackwardKernel()
163 using Input = TTypes<float, 5>::ConstTensor;
164 using Filter = TTypes<float, 5>::ConstTensor;
216 using OutputBackward = TTypes<float, 5>::ConstTensor; in CuboidConvolutionBackwardInput()
251 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 | bincount_op_gpu.cu.cc | 39 const typename TTypes<Tidx, 1>::ConstTensor& arr, in Compute()
40 const typename TTypes<T, 1>::ConstTensor& weights, in Compute()
118 const typename TTypes<Tidx, 1>::ConstTensor& arr, in Compute()
119 const typename TTypes<T, 1>::ConstTensor& weights, in Compute()
198 const typename TTypes<Tidx, 2>::ConstTensor& in, in Compute()
199 const typename TTypes<T, 2>::ConstTensor& weights, in Compute()
|
| D | searchsorted_op_gpu.cu.cc | 64 const typename TTypes<T, 1>::ConstTensor& sorted_inputs, in Compute()
65 const typename TTypes<T, 1>::ConstTensor& values, in Compute()
87 const typename TTypes<T, 1>::ConstTensor& sorted_inputs, in Compute()
88 const typename TTypes<T, 1>::ConstTensor& values, in Compute()
|
| D | tile_ops_impl.h | 30 typename TTypes<T, NDIM>::ConstTensor in, in operator()
45 typename TTypes<T, 0>::ConstTensor in,
61 typename TTypes<T, NDIM>::ConstTensor in,
|
| D | reverse_sequence_op.h | 32 ReverseGenerator(typename TTypes<T, Dims>::ConstTensor input, int32 batch_dim, in ReverseGenerator()
51 typename TTypes<T, Dims>::ConstTensor input_;
64 const Device& d, typename TTypes<T, Dims>::ConstTensor input, in Compute()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | tensor.h | 528 typename TTypes<T, NDIMS>::ConstTensor tensor() const;
536 typename TTypes<T, NDIMS>::ConstTensor bit_casted_tensor() const;
546 typename TTypes<T, NDIMS>::ConstTensor reinterpret_last_dimension() const;
559 typename TTypes<T, NDIMS>::ConstTensor shaped(
568 typename TTypes<T, NDIMS>::ConstTensor bit_casted_shaped(
579 typename TTypes<T, NDIMS>::ConstTensor flat_inner_dims() const;
582 typename TTypes<T, NDIMS>::ConstTensor flat_outer_dims() const;
585 typename TTypes<T, NDIMS>::ConstTensor flat_inner_outer_dims(
751 typename TTypes<T, NDIMS>::ConstTensor Tensor::tensor() const { in tensor()
753 return typename TTypes<T, NDIMS>::ConstTensor(base<const T>(), in tensor()
[all …]
|