| /external/tensorflow/tensorflow/core/kernels/ |
| sparse_dense_binary_op_shared.cc | 55 template <typename Device, typename T, typename Functor>
164 typename Functor::func());
176 SparseDenseBinaryOpShared<CPUDevice, T, functor::mul<T>>) \
180 SparseDenseBinaryOpShared<CPUDevice, T, functor::div<T>>) \
183 SparseDenseBinaryOpShared<CPUDevice, T, functor::add<T>>)
|
| cudnn_pooling_gpu.cc | 53 functor::NHWCToNCHW<GPUDevice, T, 5>()(context->eigen_device<GPUDevice>(),
109 functor::NCHWToNHWC<GPUDevice, T, 5>()(
175 functor::NHWCToNCHW<GPUDevice, T, 5>()(context->eigen_device<GPUDevice>(),
180 functor::NHWCToNCHW<GPUDevice, T, 5>()(context->eigen_device<GPUDevice>(),
184 functor::NHWCToNCHW<GPUDevice, T, 5>()(
239 functor::NCHWToNHWC<GPUDevice, T, 5>()(
|
| resize_nearest_neighbor_op.cc | 65 functor::ResizeNearestNeighbor<Device, T, /*align_corners=*/true>()(
70 functor::ResizeNearestNeighbor<Device, T, /*align_corners=*/false>()(
84 // Partial specialization of ResizeNearestNeighbor functor for a CPUDevice.
85 namespace functor { namespace in namespace:tensorflow
117 } // namespace functor
175 status = functor::ResizeNearestNeighborGrad<Device, T,
180 status = functor::ResizeNearestNeighborGrad<Device, T,
195 // Partial specialization of ResizeNearestNeighborGrad functor for a CPUDevice.
196 namespace functor { namespace in namespace:tensorflow
232 } // namespace functor
[all...] |
| scatter_op.cc | 127 functor::ScatterFunctor<Device, T, Index, op> functor; local
128 const Index bad_i = functor(c, c->template eigen_device<Device>(),
200 functor::ScatterFunctorSYCL<T, Index, op> functor; local
201 const Index bad_i = functor(c, c->template eigen_device<SYCLDevice>(),
|
| where_op.cc | 53 namespace functor { namespace in namespace:tensorflow
121 } // namespace functor
144 Status s = functor::NumTrue<CPUDevice, T, int64>::Compute(
160 Status s = functor::Where<CPUDevice, NDIM, T, int64>::Compute( \
204 namespace functor { namespace
247 } // namespace functor
285 Status s = functor::NumTrue<GPUDevice, T, Tindex>::Compute(
324 Status s = functor::Where<GPUDevice, NDIM, T, Tindex>::Compute( \
|
| bucketize_op_gpu.cu.cc | 77 namespace functor { namespace in namespace:tensorflow
114 } // namespace functor
117 template struct functor::BucketizeFunctor<GPUDevice, type>;
|
| cross_op.cc | 70 // while flattening all others do give the functor easy to work with data.
78 functor::Cross<Device, Type>()(context->eigen_device<Device>(), in0_data,
93 namespace functor { namespace in namespace:tensorflow
103 } // namespace functor
|
| histogram_op_gpu.cu.cc | 35 namespace functor { namespace in namespace:tensorflow
115 } // end namespace functor
118 template struct functor::HistogramFixedWidthFunctor<GPUDevice, type, int32>;
|
| nth_element_op.cc | 75 functor::NthElementFunctor<Device, T> nthElementFunc;
83 namespace functor { namespace in namespace:tensorflow
124 } // namespace functor
|
| tensor_array.cc | 34 functor::Add2Functor<Device, T> add_functor; \
59 functor::SetZeroFunctor<Device, T> set_zero_functor; \
|
| argmax_op.h | 26 namespace functor { namespace in namespace:tensorflow
64 } // namespace functor
|
| avgpooling_op.h | 18 // Functor definition for AvgPoolingOp, must be compilable by nvcc.
25 namespace functor { namespace in namespace:tensorflow
47 } // namespace functor
|
| betainc_op.h | 18 // Functor definition for BetaincOp, must be compilable by nvcc.
24 namespace functor { namespace in namespace:tensorflow
26 // Functor used by BetaincOp to do the computations.
48 } // namespace functor
|
| bias_op.h | 18 // Functor definition for BiasOp, must be compilable by nvcc.
24 namespace functor { namespace in namespace:tensorflow
26 // Functor used by BiasOp to do the computations.
52 } // namespace functor
|
| cast_op_gpu.cu.cc | 24 namespace functor { namespace in namespace:tensorflow
69 } // end namespace functor
|
| cast_op_impl_bfloat.cc | 46 functor::CastFunctor<GPUDevice, float, bfloat16> func;
|
| crop_and_resize_op.h | 25 namespace functor { namespace in namespace:tensorflow
67 } // namespace functor
|
| cwise_ops_gpu_gradients.cu.h | 34 namespace functor { namespace in namespace:tensorflow
40 // Partial specialization of SimpleBinaryFunctor<Device=GPUDevice, Functor>.
41 template <typename Functor>
42 struct SimpleBinaryFunctor<GPUDevice, Functor> {
43 void operator()(const GPUDevice& d, typename Functor::tout_type out,
44 typename Functor::tin_type in1,
45 typename Functor::tin_type in2) {
47 To32Bit(in1).binaryExpr(in2, typename Functor::func());
52 // (T0, T1, etc.) for SimpleBiaryFunctor (e.g., functor::tanh_grad).
68 } // end namespace functor
[all...] |
| deep_conv2d.h | 104 namespace functor { namespace in namespace:tensorflow
113 } // namespace functor
|
| dense_update_functor.cc | 30 namespace functor { namespace in namespace:tensorflow
71 } // namespace functor
|
| depthtospace_op_gpu.cu.cc | 145 namespace functor { namespace in namespace:tensorflow
226 } // end namespace functor
229 template struct functor::DepthToSpaceOpFunctor<GPUDevice, float, FORMAT_NCHW>;
230 template struct functor::DepthToSpaceOpFunctor<GPUDevice, float, FORMAT_NHWC>;
233 template struct functor::DepthToSpaceOpFunctor<GPUDevice, int32, FORMAT_NCHW>;
|
| /external/eigen/unsupported/Eigen/src/LevenbergMarquardt/ |
| LevenbergMarquardt.h | 89 // to be defined in the functor
92 // to be defined in the functor if no automatic differentiation
122 LevenbergMarquardt(FunctorType& functor)
123 : m_functor(functor),m_nfev(0),m_njev(0),m_fnorm(0.0),m_gnorm(0),
138 FunctorType &functor,
368 FunctorType &functor,
375 Index m = functor.values();
381 NumericalDiff<FunctorType> numDiff(functor);
|
| /frameworks/base/libs/hwui/ |
| DisplayList.cpp | 75 iter.listener->onGlFunctorReleased(iter.functor);
94 (*iter.functor)(DrawGlInfo::kModeSync, nullptr);
|
| /frameworks/base/libs/hwui/pipeline/skia/ |
| SkiaDisplayList.cpp | 31 for (auto& functor : mChildFunctors) {
32 functor.syncFunctor();
|
| /external/tensorflow/tensorflow/contrib/rnn/kernels/ |
| blas_gemm.cc | 37 namespace functor { namespace in namespace:tensorflow
67 } // end namespace functor
|
