Home
last modified time | relevance | path

Searched refs:Tout (Results 1 – 25 of 161) sorted by relevance

1234567

/external/tensorflow/tensorflow/core/kernels/
Ddebug_ops.h458 template <typename Tin, typename Tout>
460 void Run(const GPUDevice& d, const Tin* data, int size, Tout output[1]);
470 template <typename Tin, typename Tout>
472 void Run(const GPUDevice& d, const Tin* data, int size, Tout output[3]);
482 template <typename Tin, typename Tout>
484 void Run(const GPUDevice& d, const Tin* data, int size, Tout output[6]);
494 template <typename Tin, typename Tout>
496 void Run(const GPUDevice& d, const Tin* data, int size, Tout output[3]);
508 template <typename Device, typename Tin, typename Tout>
512 template <typename Tin, typename Tout>
[all …]
Ddebug_ops_gpu.cu.cc38 template <typename Tin, typename Tout>
40 Tout output[1]) { in CurtHealthKernel()
56 template <typename Tin, typename Tout>
58 Tout output[3]) { in ConciseHealthKernel()
63 Tout accum[3] = {0.0, 0.0, 0.0}; in ConciseHealthKernel()
87 template <typename Tin, typename Tout>
89 Tout output[6]) { in FullHealthKernel()
94 Tout accum[6] = {0.0, 0.0, 0.0, 0.0, 0.0, 0.0}; in FullHealthKernel()
130 template <typename Tin, typename Tout>
132 int size, Tout output[3]) { in ReduceInfNanThreeSlotsKernel()
[all …]
Dargmax_op.cc44 template <typename Device, typename T, typename Tout, typename ArgFunctor>
88 output->tensor<Tout, NDIM - 1>()); \ in Compute()
114 template <typename Device, typename T, typename Tout>
116 : public ArgOp<Device, T, Tout, functor::ArgMax<Device, T, Tout> > {
119 : ArgOp<Device, T, Tout, functor::ArgMax<Device, T, Tout> >(context) {} in ArgMaxOp()
122 template <typename Device, typename T, typename Tout>
124 : public ArgOp<Device, T, Tout, functor::ArgMin<Device, T, Tout> > {
127 : ArgOp<Device, T, Tout, functor::ArgMin<Device, T, Tout> >(context) {} in ArgMinOp()
165 #define DECLARE_GPU_SPEC(T, Tout, Dims) \ argument
167 void ArgMax<GPUDevice, T, Tout>::Reduce##Dims( \
[all …]
Dargmax_op.h28 template <typename Device, typename T, typename Tout>
33 const int32 dimension, typename TTypes<Tout, Dims - 1>::Tensor output) { \
34 output.device(d) = input.argmax(dimension).template cast<Tout>(); \
48 template <typename Device, typename T, typename Tout>
53 const int32 dimension, typename TTypes<Tout, Dims - 1>::Tensor output) { \
54 output.device(d) = input.argmin(dimension).template cast<Tout>(); \
Dhistogram_op.cc34 template <typename T, typename Tout>
35 struct HistogramFixedWidthFunctor<CPUDevice, T, Tout> {
39 int32 nbins, typename TTypes<Tout, 1>::Tensor& out) { in Compute()
70 out(index_to_bin(i)) += Tout(1); in Compute()
78 template <typename Device, typename T, typename Tout>
113 auto out = out_tensor->flat<Tout>(); in Compute()
116 ctx, functor::HistogramFixedWidthFunctor<Device, T, Tout>::Compute( in Compute()
Dcwise_ops_common.h82 typedef typename Functor::out_type Tout; // Output scalar data type. typedef
85 : BinaryOpShared(ctx, DataTypeToEnum<Tout>::v(), in BinaryOp()
103 eigen_device, out->template flat<Tout>(), in Compute()
117 eigen_device, out->template flat<Tout>(), in Compute()
130 eigen_device, out->template flat<Tout>(), in Compute()
168 auto out_flat = out->flat<Tout>(); in Compute()
186 eigen_device, out->shaped<Tout, 2>(bcast.result_shape()), in Compute()
193 eigen_device, out->shaped<Tout, 3>(bcast.result_shape()), in Compute()
200 eigen_device, out->shaped<Tout, 4>(bcast.result_shape()), in Compute()
207 eigen_device, out->shaped<Tout, 5>(bcast.result_shape()), in Compute()
[all …]
Disotonic_regression_op.cc170 template <typename Tin, typename Tout>
191 auto output = output_tensor->flat_inner_dims<Tout, 2>(); in Compute()
200 return L2PavaSegment<Tout>(input(row_index, col_index), col_index); in Compute()
202 solve_pava_batch<L2PavaSegment<Tout>>(make_l2_segment, &output, &segments, in Compute()
207 #define REGISTER_CPU_KERNEL(Tin, Tout) \ argument
211 .TypeConstraint<Tout>("output_dtype"), \
212 IsotonicRegressionOp<Tin, Tout>);
/external/tensorflow/tensorflow/python/ops/
Dscript_ops.py74 def __init__(self, func, Tout, is_grad_func, use_tape_cache=True): argument
88 self._out_dtypes = Tout
281 Tout, argument
296 if isinstance(Tout, (list, tuple)):
299 Tout = [Tout]
302 func = EagerFunc(func, Tout, is_grad_func, use_tape_cache=use_tape_cache)
347 Tout=Tout,
352 input=inp, token=token, Tout=Tout, name=name)
355 input=inp, token=token, Tout=Tout, name=name)
374 Tout=[tensor.dtype for tensor in op.inputs],
[all …]
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/
DIf.pbtxt13 type_list_attr: "Tout"
26 name: "Tout"
52 type_list_attr: "Tout"
64 name: "Tout"
90 type_list_attr: "Tout"
102 name: "Tout"
127 type_list_attr: "Tout"
139 name: "Tout"
165 type_list_attr: "Tout"
177 name: "Tout"
DPartitionedCall.pbtxt9 type_list_attr: "Tout"
17 name: "Tout"
34 type_list_attr: "Tout"
42 name: "Tout"
66 type_list_attr: "Tout"
74 name: "Tout"
105 type_list_attr: "Tout"
113 name: "Tout"
DStatefulPartitionedCall.pbtxt9 type_list_attr: "Tout"
17 name: "Tout"
35 type_list_attr: "Tout"
43 name: "Tout"
68 type_list_attr: "Tout"
76 name: "Tout"
108 type_list_attr: "Tout"
116 name: "Tout"
DRandomUniformInt.pbtxt9 type_attr: "Tout"
13 type_attr: "Tout"
17 type_attr: "Tout"
34 name: "Tout"
DEagerPyFunc.pbtxt9 type_list_attr: "Tout"
21 name: "Tout"
35 type_list_attr: "Tout"
54 name: "Tout"
DRemoteCall.pbtxt13 type_list_attr: "Tout"
22 name: "Tout"
44 type_list_attr: "Tout"
53 name: "Tout"
DTPUPartitionedCall.pbtxt13 type_list_attr: "Tout"
21 name: "Tout"
42 type_list_attr: "Tout"
50 name: "Tout"
DStatelessIf.pbtxt13 type_list_attr: "Tout"
25 name: "Tout"
50 type_list_attr: "Tout"
62 name: "Tout"
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/
DIf.pbtxt13 type_list_attr: "Tout"
26 name: "Tout"
52 type_list_attr: "Tout"
64 name: "Tout"
90 type_list_attr: "Tout"
102 name: "Tout"
127 type_list_attr: "Tout"
139 name: "Tout"
165 type_list_attr: "Tout"
177 name: "Tout"
DPartitionedCall.pbtxt9 type_list_attr: "Tout"
17 name: "Tout"
34 type_list_attr: "Tout"
42 name: "Tout"
66 type_list_attr: "Tout"
74 name: "Tout"
105 type_list_attr: "Tout"
113 name: "Tout"
DStatefulPartitionedCall.pbtxt9 type_list_attr: "Tout"
17 name: "Tout"
35 type_list_attr: "Tout"
43 name: "Tout"
68 type_list_attr: "Tout"
76 name: "Tout"
108 type_list_attr: "Tout"
116 name: "Tout"
DRandomUniformInt.pbtxt9 type_attr: "Tout"
13 type_attr: "Tout"
17 type_attr: "Tout"
34 name: "Tout"
DEagerPyFunc.pbtxt9 type_list_attr: "Tout"
21 name: "Tout"
35 type_list_attr: "Tout"
54 name: "Tout"
DRemoteCall.pbtxt13 type_list_attr: "Tout"
22 name: "Tout"
44 type_list_attr: "Tout"
53 name: "Tout"
DTPUPartitionedCall.pbtxt13 type_list_attr: "Tout"
21 name: "Tout"
42 type_list_attr: "Tout"
50 name: "Tout"
DStatelessIf.pbtxt13 type_list_attr: "Tout"
25 name: "Tout"
50 type_list_attr: "Tout"
62 name: "Tout"
DLookupTableFindV2.pbtxt13 type_attr: "Tout"
17 type_attr: "Tout"
24 name: "Tout"

1234567