Home
last modified time | relevance | path

Searched refs:Tin (Results 1 – 25 of 98) sorted by relevance

1234

/external/tensorflow/tensorflow/core/kernels/
Ddebug_ops_gpu.cu.cc38 template <typename Tin, typename Tout>
39 __global__ void CurtHealthKernel(const Tin* __restrict__ data, int size, in CurtHealthKernel()
56 template <typename Tin, typename Tout>
57 __global__ void ConciseHealthKernel(const Tin* __restrict__ data, int size, in ConciseHealthKernel()
67 if (data[offset] < static_cast<Tin>(0.f)) { in ConciseHealthKernel()
87 template <typename Tin, typename Tout>
88 __global__ void FullHealthKernel(const Tin* __restrict__ data, int size, in FullHealthKernel()
98 if (data[offset] < static_cast<Tin>(0.f)) { in FullHealthKernel()
106 if (data[offset] < static_cast<Tin>(0.f)) { in FullHealthKernel()
108 } else if (data[offset] == static_cast<Tin>(0.f)) { in FullHealthKernel()
[all …]
Dunique_op.cc150 auto Tin = input.flat<T>(); in Compute() local
151 const int64 N = static_cast<int64>(Tin.size()); in Compute()
156 auto it = uniq.emplace(Tin(i), j); in Compute()
176 auto Tin = input.shaped<T, 3>(new_sizes); in Compute() local
178 auto hash_fn = [&Tin](const Eigen::Index& key) { in Compute()
180 for (Eigen::Index i = 0; i < Tin.dimension(0); i++) { in Compute()
181 for (Eigen::Index j = 0; j < Tin.dimension(2); j++) { in Compute()
182 h = Hash64Combine(h, hash<T>{}(Tin(i, key, j))); in Compute()
188 auto equal_to_fn = [&Tin](const Eigen::Index& lhs, in Compute()
190 for (Eigen::Index i = 0; i < Tin.dimension(0); i++) { in Compute()
[all …]
Dcwise_ops_common.h81 typedef typename Functor::in_type Tin; // Input scalar data type. typedef
86 DataTypeToEnum<Tin>::v()) {} in BinaryOp()
104 input_0.template flat<Tin>(), input_1.template flat<Tin>(), in Compute()
118 input_0.template scalar<Tin>(), input_1.template flat<Tin>(), in Compute()
131 input_0.template flat<Tin>(), input_1.template scalar<Tin>(), in Compute()
172 eigen_device, out_flat, in0.template flat<Tin>(), in Compute()
173 in1.template scalar<Tin>(), error_ptr); in Compute()
177 eigen_device, out_flat, in0.template scalar<Tin>(), in Compute()
178 in1.template flat<Tin>(), error_ptr); in Compute()
181 eigen_device, out_flat, in0.template flat<Tin>(), in Compute()
[all …]
Disotonic_regression_op.cc170 template <typename Tin, typename Tout>
179 const auto input = input_tensor.flat_inner_dims<Tin, 2>(); in Compute()
207 #define REGISTER_CPU_KERNEL(Tin, Tout) \ argument
210 .TypeConstraint<Tin>("T") \
212 IsotonicRegressionOp<Tin, Tout>);
219 #define REGISTER_CPU_KERNEL_FLOAT(Tin) REGISTER_CPU_KERNEL(Tin, float) argument
224 #define REGISTER_CPU_KERNEL_DOUBLE(Tin) REGISTER_CPU_KERNEL(Tin, double) argument
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 …]
Dsegment_reduction_ops_impl.h584 template <typename Tin>
586 typename std::enable_if<std::is_same<Tin, bfloat16>::value, int>::type;
587 template <typename Tin>
589 typename std::enable_if<!std::is_same<Tin, bfloat16>::value, int>::type;
591 template <typename Tin, typename Tindex, EnableIfNotBfloat16<Tin> = 0>
593 const typename TTypes<Tin>::ConstMatrix& input_flat, Tindex index) {
597 template <typename Tin, typename Tindex, EnableIfBfloat16<Tin> = 0>
599 const typename TTypes<Tin>::ConstMatrix& input_flat, Tindex index) {
615 template <typename Tin, typename Tindex, EnableIfNotBfloat16<Tin> = 0>
617 const typename TTypes<Tin>::ConstMatrix& input_flat,
[all …]
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/
DIf.pbtxt9 type_list_attr: "Tin"
20 name: "Tin"
48 type_list_attr: "Tin"
59 name: "Tin"
86 type_list_attr: "Tin"
97 name: "Tin"
123 type_list_attr: "Tin"
134 name: "Tin"
161 type_list_attr: "Tin"
172 name: "Tin"
DPartitionedCall.pbtxt5 type_list_attr: "Tin"
12 name: "Tin"
30 type_list_attr: "Tin"
37 name: "Tin"
62 type_list_attr: "Tin"
69 name: "Tin"
101 type_list_attr: "Tin"
108 name: "Tin"
DStatefulPartitionedCall.pbtxt5 type_list_attr: "Tin"
12 name: "Tin"
31 type_list_attr: "Tin"
38 name: "Tin"
64 type_list_attr: "Tin"
71 name: "Tin"
104 type_list_attr: "Tin"
111 name: "Tin"
DEagerPyFunc.pbtxt5 type_list_attr: "Tin"
16 name: "Tin"
31 type_list_attr: "Tin"
49 name: "Tin"
DTPUPartitionedCall.pbtxt5 type_list_attr: "Tin"
16 name: "Tin"
34 type_list_attr: "Tin"
45 name: "Tin"
DRemoteCall.pbtxt9 type_list_attr: "Tin"
16 name: "Tin"
40 type_list_attr: "Tin"
47 name: "Tin"
DStatelessIf.pbtxt9 type_list_attr: "Tin"
20 name: "Tin"
46 type_list_attr: "Tin"
57 name: "Tin"
DBatchFunction.pbtxt5 type_list_attr: "Tin"
68 name: "Tin"
89 type_list_attr: "Tin"
152 name: "Tin"
DLookupTableRemoveV2.pbtxt9 type_attr: "Tin"
12 name: "Tin"
DLookupTableInsert.pbtxt10 type_attr: "Tin"
17 name: "Tin"
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/
DIf.pbtxt9 type_list_attr: "Tin"
20 name: "Tin"
48 type_list_attr: "Tin"
59 name: "Tin"
86 type_list_attr: "Tin"
97 name: "Tin"
123 type_list_attr: "Tin"
134 name: "Tin"
161 type_list_attr: "Tin"
172 name: "Tin"
DPartitionedCall.pbtxt5 type_list_attr: "Tin"
12 name: "Tin"
30 type_list_attr: "Tin"
37 name: "Tin"
62 type_list_attr: "Tin"
69 name: "Tin"
101 type_list_attr: "Tin"
108 name: "Tin"
DStatefulPartitionedCall.pbtxt5 type_list_attr: "Tin"
12 name: "Tin"
31 type_list_attr: "Tin"
38 name: "Tin"
64 type_list_attr: "Tin"
71 name: "Tin"
104 type_list_attr: "Tin"
111 name: "Tin"
DEagerPyFunc.pbtxt5 type_list_attr: "Tin"
16 name: "Tin"
31 type_list_attr: "Tin"
49 name: "Tin"
DRemoteCall.pbtxt9 type_list_attr: "Tin"
16 name: "Tin"
40 type_list_attr: "Tin"
47 name: "Tin"
DTPUPartitionedCall.pbtxt5 type_list_attr: "Tin"
16 name: "Tin"
34 type_list_attr: "Tin"
45 name: "Tin"
DStatelessIf.pbtxt9 type_list_attr: "Tin"
20 name: "Tin"
46 type_list_attr: "Tin"
57 name: "Tin"
DLookupTableRemoveV2.pbtxt9 type_attr: "Tin"
12 name: "Tin"
/external/llvm-project/pstl/test/std/algorithms/alg.modifying.operations/
Dtransform_unary.pass.cpp51 template <typename Tin, typename Tout>
57 Sequence<Tin> in(n, [](int32_t k) { return k % 5 != 1 ? 3 * k - 7 : 0; }); in test()
61 const auto flip = Complement<Tin, Tout>(1); in test()

1234