Searched defs:alg_input_idx (Results 1 – 4 of 4) sorted by relevance
172 Status AlgorithmFromInput(XlaOpKernelContext* ctx, int alg_input_idx, in AlgorithmFromInput()223 const int alg_input_idx = 3; in Compile() local292 const int alg_input_idx = 3; in Compile() local356 const int alg_input_idx = 3; in Compile() local406 const int alg_input_idx = 3; in Compile() local476 const int alg_input_idx = 3; in Compile() local
174 XlaOpKernelContext* ctx, int state_input_idx, int alg_input_idx, in CompileImpl()435 const int alg_input_idx = 1; in Compile() local
47 int counter_input_idx, int alg_input_idx) { in GetKeyCounterAlgFromInputs()
268 auto alg_input_idx = 1; in Compute() local