Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dstateful_random_ops_gpu.cu.cc121 template struct UpdateVariableAndFill_Philox< variable
123 template struct UpdateVariableAndFill_Philox< variable
125 template struct UpdateVariableAndFill_Philox< variable
127 template struct UpdateVariableAndFill_Philox< variable
131 template struct UpdateVariableAndFill_Philox< variable
135 template struct UpdateVariableAndFill_Philox< variable
139 template struct UpdateVariableAndFill_Philox< variable
141 template struct UpdateVariableAndFill_Philox< variable
143 template struct UpdateVariableAndFill_Philox< variable
145 template struct UpdateVariableAndFill_Philox< variable
[all …]
H A Dstateful_random_ops.cc31 struct UpdateVariableAndFill_Philox<CPUDevice, Distribution> { struct
32 void operator()(OpKernelContext* ctx, const CPUDevice& device, in operator ()()