Home
last modified time | relevance | path

Searched refs:ConstTensor (Results 1 – 25 of 166) sorted by relevance

1234567

/external/tensorflow/tensorflow/core/kernels/
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,
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 …]
Drelu_op_functor.h33 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 …]
Ddilation_ops.h29 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,
Drelu_op.cc83 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 …]
Dconv_2d.h34 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 …]
Dbetainc_op.h29 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()
Dscale_and_translate_op.h62 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,
Dconv_3d.h45 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,
Dmatrix_diag_op.h33 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()
Dsearchsorted_op.h33 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,
Dscatter_nd_op_cpu_impl.h96 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, \
Dbetainc_op.cc131 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, \
Dsplit_lib.h29 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,
Dcross_op.cc71 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, \
Dscale_and_translate_op.cc334 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 …]
Dsoftsign_op.h34 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()
Deigen_benchmark.h32 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()
Ddilation_ops_gpu.cu.cc181 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 ()()
Dcrop_and_resize_op_gpu.cu.cc350 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 ()()
Dsoftplus_op.h34 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()
Dsearchsorted_op_gpu.cu.cc63 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()
Dcrop_and_resize_op.cc83 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 …]
Dgather_functor.h40 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/
Dtensor.h465 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/
Dbeam_search_ops.cc83 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, \

1234567