Home
last modified time | relevance | path

Searched refs:in1_flat (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dcwise_op_clip.cc53 auto in1_flat = in1.flat<T>(); in Compute() local
60 functor::TernaryClipOp<Device, T>()(d, in0_flat, in1_flat, in2_flat, in Compute()
63 functor::UnaryClipOp<Device, T>()(d, in0_flat, in1_flat, in2_flat, in Compute()
68 functor::BinaryLeftClipOp<Device, T>()(d, in0_flat, in1_flat, in2_flat, in Compute()
71 functor::BinaryRightClipOp<Device, T>()(d, in0_flat, in1_flat, in2_flat, in Compute()
93 typename TTypes<T>::ConstFlat& in1_flat, in operator ()()
96 out_flat = in0_flat.unaryExpr(UnaryClipFunc<T>(in1_flat(0), in2_flat(0))); in operator ()()
112 typename TTypes<T>::ConstFlat& in1_flat, in operator ()()
116 in0_flat.binaryExpr(in2_flat, BinaryRightClipFunc<T>(in1_flat(0))); in operator ()()
132 typename TTypes<T>::ConstFlat& in1_flat, in operator ()()
[all …]
Dcwise_op_clip_gpu.cu.cc60 typename TTypes<T>::ConstFlat &in1_flat, in operator ()()
67 0, d.stream(), in0_flat.size(), in0_flat.data(), in1_flat.data(), in operator ()()
76 typename TTypes<T>::ConstFlat &in1_flat, in operator ()()
84 in0_flat.data(), in1_flat.data(), in2_flat.data(), out_flat.data())); in operator ()()
92 typename TTypes<T>::ConstFlat &in1_flat, in operator ()()
100 in0_flat.data(), in1_flat.data(), in2_flat.data(), out_flat.data())); in operator ()()
108 typename TTypes<T>::ConstFlat &in1_flat, in operator ()()
111 out_flat.device(d) = in0_flat.cwiseMin(in2_flat).cwiseMax(in1_flat); in operator ()()
Dcwise_op_clip.h27 typename TTypes<T>::ConstFlat &in1_flat,
36 typename TTypes<T>::ConstFlat &in1_flat,
45 typename TTypes<T>::ConstFlat &in1_flat,
54 typename TTypes<T>::ConstFlat &in1_flat,
Dcwise_ops_common.h208 auto in1_flat = in1.flat<Tin>(); in Compute() local
220 in0_flat, in1_flat); in Compute()