Searched refs:IsTensorListInput (Results 1 – 3 of 3) sorted by relevance
32 bool IsTensorListInput(XlaOpKernelContext* ctx, int index);
27 bool IsTensorListInput(XlaOpKernelContext* ctx, int index) { in IsTensorListInput() function
88 if (IsTensorListInput(ctx, i)) { in MakeXlaCompilerArgumentsFromInputs()392 if (arg.initialized || !IsTensorListInput(ctx, i)) { in Compile()482 } else if (IsTensorListInput(ctx, input_num)) { in Compile()