Searched defs:alg_input_idx (Results 1 – 4 of 4) sorted by relevance
170 Status GetAlgorithm(XlaOpKernelContext* ctx, int alg_input_idx, in GetAlgorithm()210 const int alg_input_idx = 3; in Compile() local263 const int alg_input_idx = 3; in Compile() local324 const int alg_input_idx = 3; in Compile() local370 const int alg_input_idx = 3; in Compile() local424 const int alg_input_idx = 3; in Compile() local
173 XlaOpKernelContext* ctx, int state_input_idx, int alg_input_idx, in CompileImpl()434 const int alg_input_idx = 1; in Compile() local
76 const int alg_input_idx = 3; in Compute() local
268 auto alg_input_idx = 1; in Compute() local