Searched defs:program_key (Results 1 – 2 of 2) sorted by relevance
166 auto program_key = builder.create<mlir::TF::CaseOp>( in CreateSendRecvOpsToTransferProgramKey() local181 mlir::Value program_key; member196 mlir::Value program_key; in SendRecvCompilationKey() local
103 mlir::Value program_key = GetOrCreateCompilationKey(dtensor_send); in LowerDTensorSendToXlaOp() local175 auto program_key = GetOrCreateCompilationKey(dtensor_recv); in LowerDTensorRecvToXlaOp() local