Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Dcwise_op_clip_gpu.cu.cc59 void operator()(const GPUDevice &d, typename TTypes<T>::ConstFlat &in0_flat, in operator ()()
63 CudaLaunchConfig config = GetCudaLaunchConfig(in0_flat.size(), d); in operator ()()
67 0, d.stream(), in0_flat.size(), in0_flat.data(), in1_flat.data(), in operator ()()
75 void operator()(const GPUDevice &d, typename TTypes<T>::ConstFlat &in0_flat, in operator ()()
79 CudaLaunchConfig config = GetCudaLaunchConfig(in0_flat.size(), d); in operator ()()
83 config.thread_per_block, 0, d.stream(), in0_flat.size(), in operator ()()
84 in0_flat.data(), in1_flat.data(), in2_flat.data(), out_flat.data())); in operator ()()
91 void operator()(const GPUDevice &d, typename TTypes<T>::ConstFlat &in0_flat, in operator ()()
95 CudaLaunchConfig config = GetCudaLaunchConfig(in0_flat.size(), d); in operator ()()
99 config.thread_per_block, 0, d.stream(), in0_flat.size(), in operator ()()
[all …]
Dcwise_op_clip.cc52 auto in0_flat = in0.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()
92 void operator()(const CPUDevice& d, typename TTypes<T>::ConstFlat& in0_flat, in operator ()()
96 out_flat = in0_flat.unaryExpr(UnaryClipFunc<T>(in1_flat(0), in2_flat(0))); in operator ()()
111 void operator()(const CPUDevice& d, typename TTypes<T>::ConstFlat& in0_flat, in operator ()()
116 in0_flat.binaryExpr(in2_flat, BinaryRightClipFunc<T>(in1_flat(0))); in operator ()()
131 void operator()(const CPUDevice& d, typename TTypes<T>::ConstFlat& in0_flat, in operator ()()
[all …]
Dcwise_op_clip.h26 void operator()(const Device &d, typename TTypes<T>::ConstFlat &in0_flat,
35 void operator()(const Device &d, typename TTypes<T>::ConstFlat &in0_flat,
44 void operator()(const Device &d, typename TTypes<T>::ConstFlat &in0_flat,
53 void operator()(const Device &d, typename TTypes<T>::ConstFlat &in0_flat,
Dcwise_ops_common.h207 auto in0_flat = in0.flat<Tin>(); in Compute() local
220 in0_flat, in1_flat); in Compute()