Searched defs:execute_arg_index (Results 1 – 1 of 1) sorted by relevance
124 TF::TPUGetLayoutOp BuildGetLayout(const int64_t execute_arg_index, in BuildGetLayout()150 void HandleInput(Value input, const int64_t execute_arg_index, in HandleInput()166 const int64_t execute_arg_index, Value compilation_key, in HandleReplicatedInputs()227 const int64_t execute_arg_index = input_and_idx.index(); in HandleCompileAndExecutes() local