Searched refs:program_key (Results 1 – 2 of 2) sorted by relevance
166 auto program_key = builder.create<mlir::TF::CaseOp>( in CreateSendRecvOpsToTransferProgramKey() local173 *program_key_output = program_key.getResult(0); in CreateSendRecvOpsToTransferProgramKey()181 mlir::Value program_key; member196 mlir::Value program_key; in SendRecvCompilationKey() local200 num_send_recv, &program_key))) in SendRecvCompilationKey()203 info.program_key = program_key; in SendRecvCompilationKey()343 mlir::Value device_program_key = compilation_key_recv_info[0].program_key; in HandleCompilationOps()352 mlir::Value host_program_key = compilation_key_recv_info[1].program_key; in HandleCompilationOps()
103 mlir::Value program_key = GetOrCreateCompilationKey(dtensor_send); in LowerDTensorSendToXlaOp() local128 loc, value_to_send, program_key, device_ordinal, dtensor_send.key()); in LowerDTensorSendToXlaOp()175 auto program_key = GetOrCreateCompilationKey(dtensor_recv); in LowerDTensorRecvToXlaOp() local179 /*dynamic_key=*/program_key, device_ordinal, dtensor_recv.keyAttr()); in LowerDTensorRecvToXlaOp()