Searched defs:state_input_idx (Results 1 – 2 of 2) sorted by relevance
86 OpKernelContext* ctx, Distribution dist, int state_input_idx, in UpdateVariableAndFill()129 int state_input_idx, int shape_input_idx, in StatefulRandomCompute()267 auto state_input_idx = 0; in Compute() local
174 XlaOpKernelContext* ctx, int state_input_idx, int alg_input_idx, in CompileImpl()434 const int state_input_idx = 0; in Compile() local