| /external/tensorflow/tensorflow/core/kernels/ |
| determinant_op_gpu.cu.cc | 29 namespace functor { namespace in namespace:tensorflow
165 } // namespace functor
|
| dilation_ops_gpu.cu.cc | 177 namespace functor { namespace in namespace:tensorflow
288 } // namespace functor
291 template struct functor::Dilation<GPUDevice, T>; \
292 template struct functor::DilationBackpropInput<GPUDevice, T>; \
293 template struct functor::DilationBackpropFilter<GPUDevice, T>;
|
| extract_image_patches_op.cc | 107 functor::ExtractImagePatchesForward<Device, T>()(
135 // Forward declarations of the functor specializations for GPU.
136 namespace functor { namespace in namespace:tensorflow
151 } // namespace functor
|
| fake_quant_ops.cc | 78 FakeQuantWithMinMaxArgsFunctor<Device> functor; local
79 functor(context->eigen_device<Device>(), input.flat<float>(), min_, max_,
127 FakeQuantWithMinMaxArgsGradientFunctor<Device> functor; local
128 functor(context->eigen_device<Device>(), gradient.flat<float>(),
149 // Forward declarations for functor specializations for GPU.
199 FakeQuantWithMinMaxVarsFunctor<Device> functor; variable
200 functor(context->eigen_device<Device>(), input.flat<float>(),
250 FakeQuantWithMinMaxVarsGradientFunctor<Device> functor; variable
251 functor(context->eigen_device<Device>(), gradient.flat<float>(),
335 FakeQuantWithMinMaxVarsPerChannelFunctor<Device> functor; variable
394 FakeQuantWithMinMaxVarsPerChannelGradientFunctor<Device> functor; variable
[all...] |
| gather_nd_op.cc | 50 c, functor::DoGatherNd<Device, T, Index>(c, params, indices, &out));
80 namespace functor { namespace in namespace:tensorflow
168 functor::GatherNdSlice<Device, T, Index, IXDIM> func; \
202 } // namespace functor
205 // Forward declarations of the functor specializations for GPU.
206 namespace functor { namespace in namespace:tensorflow
237 } // namespace functor
|
| matrix_band_part_op.cc | 103 functor::MatrixBandPartFunctor<Device, T> fn;
129 // Implementation of the functor specialization for CPU.
130 namespace functor { namespace in namespace:tensorflow
196 } // namespace functor
200 // Forward declarations of the functor specializations for GPU.
201 namespace functor { namespace in namespace:tensorflow
217 } // namespace functor
|
| maxpooling_op_gpu.cu.cc | 375 namespace functor { namespace in namespace:tensorflow
513 } // namespace functor
|
| mirror_pad_op.cc | 124 functor::MirrorPad<Device, T, Tpaddings, i>()( \
152 namespace functor { namespace in namespace:tensorflow
153 // Forward declarations of the functor specializations defined in the sharded
179 } // namespace functor
200 namespace functor { namespace in namespace:tensorflow
201 // Forward declarations of the functor specializations for GPU.
225 } // namespace functor
333 functor::MirrorPadGrad<Device, T, Tpaddings, k>()( \
359 namespace functor { namespace in namespace:tensorflow
360 // Forward declarations of the functor specializations defined in the sharde
406 namespace functor { namespace in namespace:tensorflow
[all...] |
| one_hot_op.cc | 116 functor::OneHot<Device, T, TI>::Compute(ctx->eigen_device<Device>(),
145 // Forward declarations of the functor specializations for GPU.
146 namespace functor { namespace in namespace:tensorflow
168 } // namespace functor
|
| parameterized_truncated_normal_op.cc | 45 namespace functor { namespace in namespace:tensorflow
237 } // namespace functor
347 auto truncFunctor = functor::TruncatedNormalFunctor<Device, T>();
|
| parameterized_truncated_normal_op_gpu.cu.cc | 45 namespace functor { namespace in namespace:tensorflow
220 } // namespace functor
|
| pooling_ops_common.cc | 126 // Forward declarations of the functor specializations for GPU.
127 namespace functor { namespace in namespace:tensorflow
138 } // namespace functor
169 functor::NHWCToNCHW<GPUDevice, T, 4>()(context->eigen_device<Device>(),
230 functor::NCHWToNHWC<GPUDevice, T, 4>()(
312 functor::NHWCToNCHW<GPUDevice, T, 4>()(context->eigen_device<Device>(),
320 functor::NHWCToNCHW<GPUDevice, T, 4>()(context->eigen_device<Device>(),
324 functor::NHWCToNCHW<GPUDevice, T, 4>()(
382 functor::NCHWToNHWC<GPUDevice, T, 4>()(
|
| random_poisson_op.cc | 68 namespace functor { namespace in namespace:tensorflow
276 } // namespace functor
308 functor::PoissonFunctor<CPUDevice, T, U>()(
|
| 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...] |
| resize_nearest_neighbor_op_gpu.cu.cc | 95 namespace functor { namespace in namespace:tensorflow
97 // Partial specialization of ResizeNearestNeighbor functor for a GPUDevice.
131 // Partial specialization of ResizeNearestNeighborGrad functor for a GPUDevice.
173 } // namespace functor
|
| 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>(),
|
| segment_reduction_ops_gpu.cu.cc | 125 namespace functor { namespace in namespace:tensorflow
214 GPUDevice, T, Index, functor::Lowest<T>, functor::MaxOpGpu<T>>; \
216 GPUDevice, T, Index, functor::Highest<T>, functor::MinOpGpu<T>>; \
217 template struct UnsortedSegmentFunctor<GPUDevice, T, Index, functor::One<T>, \
218 functor::ProdOpGpu<T>>;
223 GPUDevice, T, Index, functor::Zero<T>, functor::SumOpGpu<T>>;
247 } // namespace functor
[all...] |
| spacetodepth_op.cc | 122 functor::SpaceToDepthOpFunctor<GPUDevice, int32, FORMAT_NCHW> functor; variable
123 functor(context->eigen_device<GPUDevice>(), Tinput_v, block_size_,
127 functor::SpaceToDepthOpFunctor<GPUDevice, T, FORMAT_NCHW> functor; variable
128 functor(context->eigen_device<GPUDevice>(), Tinput, block_size_,
138 functor::SpaceToDepthOpFunctor<Device, T, FORMAT_NHWC> functor; variable
139 functor(context->eigen_device<Device>(), Tinput, block_size_, Toutput);
149 namespace functor { namespace in namespace:tensorflow
[all...] |
| sparse_xent_op.h | 18 // Functor definition for SparseXentOp, must be compilable by nvcc.
129 namespace functor { namespace in namespace:tensorflow
131 // Functor used by SparseXentOp to do the computations.
148 // This code works for both CPU and GPU and is used by the functor
223 } // namespace functor
|
| split_lib_gpu.cu.cc | 30 namespace functor { namespace in namespace:tensorflow
67 } // namespace functor
|
| strided_slice_op.cc | 142 MemCpyFunctor<T> functor; variable
143 if (functor.Copy(input, begin, end, result)) {
|
| topk_op.cc | 90 Status s = functor::TopKFunctor<Device, T>::Compute(
100 namespace functor { namespace in namespace:tensorflow
230 } // namespace functor
247 namespace functor { namespace in namespace:tensorflow
255 extern template struct functor::TopKFunctor<GPUDevice, T>;
262 } // namespace functor
|
| 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( \
|
| /external/webrtc/webrtc/base/ |
| bind_unittest.cc | 168 auto functor = Bind(&LifeTimeCheck::NullaryVoid, &object); local
184 auto functor = Bind(&LifeTimeCheck::NullaryVoid, scoped_object); local
192 // Test Bind where method object is captured as scoped_refptr<> and the functor
213 auto functor = local
234 auto functor = Bind(&Ref, x); local
235 EXPECT_NE(&x, functor());
|
| /external/eigen/unsupported/Eigen/CXX11/src/Tensor/ |
| TensorEvaluator.h | 242 : m_functor(op.functor()), m_argImpl(op.nestedExpression(), device), m_wrapper()
279 NullaryOp functor() const { return m_functor; } function in struct:Eigen::TensorEvaluator
306 : m_functor(op.functor()),
349 UnaryOp functor() const { return m_functor; } function in struct:Eigen::TensorEvaluator
375 : m_functor(op.functor()),
430 BinaryOp functor() const { return m_functor; } function in struct:Eigen::TensorEvaluator
455 : m_functor(op.functor()),
|
