Searched refs:in0_flat (Results 1 – 4 of 4) sorted by relevance
/external/tensorflow/tensorflow/core/kernels/ |
D | cwise_op_clip_gpu.cu.cc | 59 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 …]
|
D | cwise_op_clip.cc | 52 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 …]
|
D | cwise_op_clip.h | 26 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,
|
D | cwise_ops_common.h | 207 auto in0_flat = in0.flat<Tin>(); in Compute() local 220 in0_flat, in1_flat); in Compute()
|