Home
last modified time | relevance | path

Searched defs:FillPhiloxRandomKernel (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Drandom_op.cc576 struct FillPhiloxRandomKernel<Distribution, false> { struct
577 typedef typename Distribution::ResultElementType T;
578 using write_accessor = sycl::accessor<uint8_t, 1, sycl::access::mode::write,
581 FillPhiloxRandomKernel(write_accessor& data, random::PhiloxRandom& gen, in FillPhiloxRandomKernel() argument
585 void operator()(sycl::nd_item<1> item) { in operator ()()
617 write_accessor data_;
618 random::PhiloxRandom gen_;
619 Distribution dist_;
623 struct FillPhiloxRandomKernel<Distribution, true> { struct
624 typedef typename Distribution::ResultElementType T;
[all …]