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