Home	Sort by relevance Sort by last modified time

Full Search Definition Symbol File Path History   | | Help

    Searched refs:eigen_device (Results 1 - 25 of 136) sorted by null

1 2 3 4 5 6

/external/tensorflow/tensorflow/core/kernels/
aggregate_ops.cc	89 functor2(ctx->template eigen_device<Device>(), To, I(0), I(1)); 97 functor2(ctx->template eigen_device<Device>(), To, I(0), I(1)); 102 functor3(ctx->template eigen_device<Device>(), To, I(0), I(1), I(2)); 107 functor4(ctx->template eigen_device<Device>(), To, I(0), I(1), I(2), 113 functor5(ctx->template eigen_device<Device>(), To, I(0), I(1), I(2), 119 functor6(ctx->template eigen_device<Device>(), To, I(0), I(1), I(2), 125 functor7(ctx->template eigen_device<Device>(), To, I(0), I(1), I(2), 131 functor8(ctx->template eigen_device<Device>(), To, I(0), I(1), I(2), 138 functor9(ctx->template eigen_device<Device>(), To, I(0), I(1), I(2), 147 functor8p(ctx->template eigen_device<Device>(), To, I(r), I(r + 1)     [all...]
relu_op.h	40 functor(context->eigen_device<Device>(), input.flat<T>(), 88 functor(context->eigen_device<Device>(), g.flat<T>(), a.flat<T>(), 99 functor(context->eigen_device<Device>(), input.flat<T>(), 130 functor(context->eigen_device<Device>(), g.flat<T>(), a.flat<T>(), 141 functor(context->eigen_device<Device>(), input.flat<T>(), 172 functor(context->eigen_device<Device>(), g.flat<T>(), a.flat<T>(), 183 functor(context->eigen_device<Device>(), input.flat<T>(), 214 functor(context->eigen_device<Device>(), g.flat<T>(), a.flat<T>(),
conditional_accumulator.h	92 accum_grad_->flat<T>().device(ctx->template eigen_device<Device>()) = 98 accum_grad_->flat<T>().device(ctx->template eigen_device<Device>()) += 107 ctx->template eigen_device<Device>()) =
l2loss_op.cc	44 const CPUDevice& d = context->eigen_device<CPUDevice>();
snapshot_op.h	44 const Device& device = context->eigen_device<Device>();
quantization_utils_test.cc	34 void TestRequantizeMany(Eigen::ThreadPoolDevice* eigen_device, float input_min, 52 if (eigen_device == nullptr) { 59 *eigen_device, i_tensor, input_min, input_max, output_min, output_max, 112 // If eigen_device is NULL, then the reference implementation is tested. 114 Eigen::ThreadPoolDevice* eigen_device) { 137 TestRequantizeMany(eigen_device, r[0], r[1], r[2], r[3], 152 TestRequantizeMany(eigen_device, -1.0f, 1.0f, -1.0f, 1.0f, vals); 153 TestRequantizeMany(eigen_device, -255.0f, 255.0f, -255.0f, 255.0f, vals); 154 TestRequantizeMany(eigen_device, -1.0f, 1.0f, -12345678.0f, 12345678.0f, 156 TestRequantizeMany(eigen_device, -1.0f, 12345678.0f, -12345678.0f     [all...]
cudnn_pooling_gpu.cc	53 functor::NHWCToNCHW<GPUDevice, T, 5>()(context->eigen_device<GPUDevice>(), 110 context->eigen_device<GPUDevice>(), 175 functor::NHWCToNCHW<GPUDevice, T, 5>()(context->eigen_device<GPUDevice>(), 180 functor::NHWCToNCHW<GPUDevice, T, 5>()(context->eigen_device<GPUDevice>(), 185 context->eigen_device<GPUDevice>(), out_backprop.tensor<T, 5>(), 240 context->eigen_device<GPUDevice>(),
transpose_op.cc	208 return ::tensorflow::DoTranspose(ctx->eigen_device<CPUDevice>(), in, perm, 217 return ::tensorflow::DoConjugateTranspose(ctx->eigen_device<CPUDevice>(), in, 257 return ::tensorflow::DoTranspose(ctx->eigen_device<GPUDevice>(), in, perm, 265 return ::tensorflow::DoConjugateTranspose(ctx->eigen_device<GPUDevice>(), in, 288 return ::tensorflow::DoTranspose(ctx->eigen_device<SYCLDevice>(), in, perm, 296 return ::tensorflow::DoConjugateTranspose(ctx->eigen_device<SYCLDevice>(), in,
diag_op_gpu.cu.cc	61 const GPUDevice& device = context->eigen_device<GPUDevice>(); 100 const GPUDevice& device = context->eigen_device<GPUDevice>();
tensor_array.cc	35 add_functor(ctx->template eigen_device<Device>(), sum->flat<T>(), \ 60 set_zero_functor(ctx->template eigen_device<Device>(), value->flat<T>()); \
cast_op_impl_bfloat.cc	47 func(ctx->eigen_device<GPUDevice>(), out->flat<float>(),
quantize_op.cc	155 o.device(ctx->template eigen_device<Device>()) = 179 ctx->template eigen_device<Device>(), input, min_range, max_range, 208 o.device(ctx->template eigen_device<Device>()) = 215 o.device(ctx->template eigen_device<Device>()) =
fused_batch_norm_op.cc	73 const CPUDevice& d = context->eigen_device<CPUDevice>(); 162 const CPUDevice& d = context->eigen_device<CPUDevice>(); 246 f(context->eigen_device<GPUDevice>(), batch_mean->flat<U>()); 247 f(context->eigen_device<GPUDevice>(), batch_var->flat<U>()); 265 context->eigen_device<GPUDevice>(), 310 GPUDevice d = context->eigen_device<GPUDevice>(); 355 context->eigen_device<GPUDevice>(), 405 context->eigen_device<GPUDevice>(), 417 context->eigen_device<GPUDevice>(), 478 context->eigen_device<GPUDevice>()     [all...]
cwise_ops_common.h	101 const Device& eigen_device = ctx->eigen_device<Device>(); variable 109 eigen_device, out_flat, in0.template flat<Tin>(), 114 eigen_device, out_flat, in0.template scalar<Tin>(), 118 eigen_device, out_flat, in0.template flat<Tin>(), 123 eigen_device, out->shaped<Tout, 2>(bcast->result_shape()), 130 eigen_device, out->shaped<Tout, 3>(bcast->result_shape()), 137 eigen_device, out->shaped<Tout, 4>(bcast->result_shape()), 144 eigen_device, out->shaped<Tout, 5>(bcast->result_shape()), 178 const Device& d = context->eigen_device<Device>() 207 const Device& eigen_device = ctx->eigen_device<Device>(); variable     [all...]
bias_op.cc	138 const Device& d = context->eigen_device<Device>(); 170 functor(ctx->eigen_device<Device>(), input.tensor<T, Dims>(), bias.vec<T>(), 258 output->template flat<T>().device(context->eigen_device<Device>()) = 271 output->template flat<T>().device(context->eigen_device<Device>()) = 343 BiasGPU<T>::compute(context->template eigen_device<Device>(), 400 BiasGradGPU<T>::compute(context->template eigen_device<Device>(),
fake_quant_ops.cc	79 functor(context->eigen_device<Device>(), input.flat<float>(), min_, max_, 128 functor(context->eigen_device<Device>(), gradient.flat<float>(), 200 functor(context->eigen_device<Device>(), input.flat<float>(), 251 functor(context->eigen_device<Device>(), gradient.flat<float>(), 336 functor(context->eigen_device<Device>(), input.flat_inner_dims<float, 2>(), 396 context->eigen_device<Device>(), gradient.flat_inner_dims<float, 2>(),
pooling_ops_common.cc	169 functor::NHWCToNCHW<GPUDevice, T, 4>()(context->eigen_device<Device>(), 231 context->eigen_device<Device>(), 312 functor::NHWCToNCHW<GPUDevice, T, 4>()(context->eigen_device<Device>(), 320 functor::NHWCToNCHW<GPUDevice, T, 4>()(context->eigen_device<Device>(), 325 context->eigen_device<Device>(), out_backprop.tensor<T, 4>(), 383 context->eigen_device<Device>(),
colorspace_op.cc	76 functor::RGBToHSV<Device, T>()(context->eigen_device<Device>(), input_data, 105 functor::HSVToRGB<Device, T>()(context->eigen_device<Device>(), input_data,
data_format_ops.cc	56 functor::DataFormatDimMap<Device, T>()(context->eigen_device<Device>(), 108 context->eigen_device<Device>(), input.flat<T>(), output->flat<T>(),
dense_update_ops.cc	40 copy(context->eigen_device<Device>(), lhs->flat<T>(), rhs.flat<T>()); 81 update_functor(context->template eigen_device<Device>(), Tparams.flat<T>(),
mkl_transpose_op.cc	114 return ::tensorflow::DoTranspose(ctx->eigen_device<CPUDevice>(), in, perm, 142 return ::tensorflow::DoConjugateTranspose(ctx->eigen_device<CPUDevice>(), in,
softplus_op.cc	44 functor(context->eigen_device<Device>(), input.flat<T>(), 80 functor(context->eigen_device<Device>(), g.flat<T>(), a.flat<T>(),
softsign_op.cc	44 functor(context->eigen_device<Device>(), input.flat<T>(), 81 functor(context->eigen_device<Device>(), g.flat<T>(), a.flat<T>(),
sparse_softmax_op.cc	97 const Device &device = context->eigen_device<Device>(); 103 tmp_scalar.device(context->eigen_device<Device>()) = group_vals.maximum();
/external/tensorflow/tensorflow/contrib/image/kernels/
segmentation_ops.cc	57 TensorRangeFunctor<Device>()(ctx->eigen_device<Device>(), 60 rank.device(ctx->eigen_device<Device>()) = rank.constant(OutputType(0)); 116 FindRootFunctor<CPUDevice, T>()(ctx->eigen_device<CPUDevice>(), output,

Completed in 969 milliseconds

1 2 3 4 5 6