| /external/tensorflow/tensorflow/core/kernels/ |
| scatter_functor.h | 125 namespace functor { namespace in namespace:tensorflow
244 } // namespace functor
|
| scatter_nd_op_cpu_impl.h | 19 // Functor definitions for ScatterND ops, must be compilable by nvcc.
87 namespace functor { namespace in namespace:tensorflow
89 // Implementation of update functor for CPU.
171 // Implementation of update functor for SYCL.
256 } // namespace functor
|
| scatter_nd_op_gpu.cu.cc | 108 namespace functor { namespace in namespace:tensorflow
110 // Functor used by ScatterOp to do the computations.
150 } // namespace functor
153 template struct functor::ScatterNdFunctor<GPUDevice, T, Index, op, IXDIM>;
|
| spacetobatch_functor.cc | 28 namespace functor { namespace in namespace:tensorflow
167 } // namespace functor
|
| spacetobatch_functor_gpu.cu.cc | 100 namespace functor { namespace in namespace:tensorflow
166 } // end namespace functor
|
| sparse_tensor_dense_add_op.cc | 96 functor::ScatterNdFunctor<Device, T, Index, N, \
123 namespace functor { namespace in namespace:tensorflow
144 } // namespace functor
|
| sparse_tensor_dense_matmul_op_gpu.cu.cc | 63 namespace functor { namespace in namespace:tensorflow
93 } // namespace functor
96 template struct functor::SparseTensorDenseMatMulFunctor< \
98 template struct functor::SparseTensorDenseMatMulFunctor< \
100 template struct functor::SparseTensorDenseMatMulFunctor< \
102 template struct functor::SparseTensorDenseMatMulFunctor< \
|
| tile_functor_gpu.cu.cc | 84 namespace functor { namespace in namespace:tensorflow
88 // Register functors used for Tile functor.
105 } // end namespace functor
|
| training_ops_gpu.cu.cc | 27 namespace functor { namespace in namespace:tensorflow
259 } // namespace functor
261 template struct functor::ApplyGradientDescent<GPUDevice, Eigen::half>;
262 template struct functor::ApplyGradientDescent<GPUDevice, float>;
263 template struct functor::ApplyGradientDescent<GPUDevice, double>;
265 template struct functor::ApplyAdagrad<GPUDevice, Eigen::half>;
266 template struct functor::ApplyAdagrad<GPUDevice, float>;
267 template struct functor::ApplyAdagrad<GPUDevice, double>;
269 template struct functor::ApplyAdadelta<GPUDevice, Eigen::half>;
270 template struct functor::ApplyAdadelta<GPUDevice, float>
[all...] |
| batch_norm_op.cc | 74 functor::BatchNorm<Device, T>()(
150 functor::BatchNormGrad<Device, T>()(
175 // Forward declarations of the functor specializations for GPU.
176 namespace functor { namespace in namespace:tensorflow
192 } // namespace functor
231 // Forward declarations of the functor specializations for GPU.
232 namespace functor { namespace in namespace:tensorflow
251 } // namespace functor
|
| bias_op.cc | 169 functor::Bias<Device, T, Dims> functor; local
170 functor(ctx->eigen_device<Device>(), input.tensor<T, Dims>(), bias.vec<T>(),
|
| compare_and_bitpack_op.cc | 70 functor::CompareAndBitpack<Device, T> func;
85 namespace functor { namespace in namespace:tensorflow
165 } // namespace functor
179 namespace functor { namespace in namespace:tensorflow
194 } // namespace functor
|
| constant_op.cc | 197 functor::FillFunctor<Device, T> functor; variable
198 functor(context->eigen_device<Device>(), out->flat<T>(),
296 functor::SetZeroFunctor<Device, T> f;
353 functor::SetOneFunctor<Device, T> f;
|
| cwise_op_select.cc | 104 functor::BatchSelectFunctor<Device, T> func;
118 functor::SelectFunctor<Device, T> func;
138 functor::SelectScalarFunctor<Device, T> func;
190 namespace functor { namespace in namespace:tensorflow
266 } // namespace functor
|
| cwise_ops_sycl_common.h | 30 namespace functor { namespace in namespace:tensorflow
39 // Partial specialization of UnaryFunctor<Device=SYCLDevice, Functor>.
40 template <typename Functor>
41 struct UnaryFunctor<SYCLDevice, Functor> {
42 void operator()(const SYCLDevice& d, typename Functor::tout_type out,
43 typename Functor::tin_type in) {
44 To32Bit(out).device(d) = To32Bit(in).unaryExpr(typename Functor::func());
48 // Partial specialization of BinaryFunctor<Device=SYCLDevice, Functor>.
49 template <typename Functor, int NDIMS, bool has_errors>
50 struct BinaryFunctor<SYCLDevice, Functor, NDIMS, has_errors>
[all...] |
| depthtospace_op.cc | 120 functor::DepthToSpaceOpFunctor<GPUDevice, int32, FORMAT_NCHW> functor; variable
121 functor(context->eigen_device<GPUDevice>(), Tinput_v, block_size_,
125 functor::DepthToSpaceOpFunctor<GPUDevice, T, FORMAT_NCHW> functor; variable
126 functor(context->eigen_device<GPUDevice>(), Tinput, block_size_,
136 functor::DepthToSpaceOpFunctor<Device, T, FORMAT_NHWC> functor; variable
137 functor(context->eigen_device<Device>(), Tinput, block_size_, Toutput);
148 namespace functor { namespace in namespace:tensorflow
[all...] |
| determinant_op.cc | 161 functor::SetOneFunctor<GPUDevice, Scalar> f;
236 functor::DeterminantFromPivotedLUFunctor<GPUDevice, Scalar> functor; variable
237 functor(d,
304 functor::SetOneFunctor<GPUDevice, Scalar> one_func;
306 functor::SetZeroFunctor<GPUDevice, Scalar> zero_func;
385 functor::LogDeterminantFromPivotedLUFunctor<GPUDevice, Scalar> functor; variable
386 functor(d, input_copy_reshaped_const, pivots_mat.data(), sign_reshaped,
|
| /external/v8/tools/testrunner/local/ |
| progress.py | 102 def functor(self, *args, **kwargs): function in function:.Register.wrap_functor
106 return functor
|
| /external/eigen/Eigen/src/Core/ |
| CwiseBinaryOp.h | 62 * \tparam BinaryOp template functor implementing the operator
87 typedef typename internal::remove_all<BinaryOp>::type Functor;
136 /** \returns the functor representing the binary operation */
138 const BinaryOp& functor() const { return m_functor; } function in class:Eigen::CwiseBinaryOp
|
| CwiseNullaryOp.h | 29 * \brief Generic expression of a matrix where all coefficients are defined by a functor
31 * \tparam NullaryOp template functor implementing the operator
41 * The functor NullaryOp must expose one of the following method:
82 /** \returns the functor representing the nullary operation */
84 const NullaryOp& functor() const { return m_functor; } function in class:Eigen::CwiseNullaryOp
93 /** \returns an expression of a matrix defined by a custom functor \a func
102 * The template parameter \a CustomNullaryOp is the type of the functor.
114 /** \returns an expression of a matrix defined by a custom functor \a func
125 * The template parameter \a CustomNullaryOp is the type of the functor.
142 /** \returns an expression of a matrix defined by a custom functor \a fun
[all...] |
| CwiseTernaryOp.h | 61 * \tparam TernaryOp template functor implementing the operator
173 /** \returns the functor representing the ternary operation */
175 const TernaryOp& functor() const { return m_functor; } function in class:Eigen::CwiseTernaryOp
|
| /external/tensorflow/tensorflow/contrib/image/kernels/ |
| image_ops.h | 142 // NOTE(ringwalt): We MUST wrap the generate() call in a functor and explicitly
143 // instantiate the functor in image_ops_gpu.cu.cc. Otherwise, we will be missing
145 namespace functor { namespace in namespace:tensorflow
169 } // end namespace functor
|
| /external/tensorflow/tensorflow/contrib/resampler/kernels/ |
| resampler_ops.cc | 36 namespace functor { namespace in namespace:tensorflow
123 } // namespace functor
171 functor::Resampler2DFunctor<Device, T>()(
202 namespace functor { namespace in namespace:tensorflow
327 } // namespace functor
381 functor::ResamplerGrad2DFunctor<Device, T>()(
|
| /external/tensorflow/tensorflow/contrib/rnn/kernels/ |
| gru_ops.cc | 150 functor::GRUBlockCellFprop<Device, T, USE_CUBLAS>(batch_size, input_size,
358 functor::GRUBlockCellBprop<Device, T, USE_CUBLAS>(batch_size, input_size,
386 // Forward declare the GPU Fprop functor.
387 namespace functor { namespace in namespace:tensorflow
405 } // end namespace functor
416 // Forward declare the GPU Bprop functor.
417 namespace functor { namespace in namespace:tensorflow
438 } // end namespace functor
|
| lstm_ops_gpu.cu.cc | 28 namespace functor { namespace in namespace:tensorflow
409 } // end namespace functor
|
