Searched refs:program_key (Results 1 – 10 of 10) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/ |
| H A D | move_compilation_to_host.cc | 166 auto program_key = builder.create<mlir::TF::CaseOp>( in CreateSendRecvOpsToTransferProgramKey() local
173 *program_key_output = program_key.getResult(0); in CreateSendRecvOpsToTransferProgramKey()
181 mlir::Value program_key; member
196 mlir::Value program_key; in SendRecvCompilationKey() local
200 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()
|
| H A D | dtensor_send_recv.cc | 103 mlir::Value program_key = GetOrCreateCompilationKey(dtensor_send); in LowerDTensorSendToXlaOp() local
128 loc, value_to_send, program_key, device_ordinal, dtensor_send.key()); in LowerDTensorSendToXlaOp()
175 auto program_key = GetOrCreateCompilationKey(dtensor_recv); in LowerDTensorRecvToXlaOp() local
179 /*dynamic_key=*/program_key, device_ordinal, dtensor_recv.keyAttr()); in LowerDTensorRecvToXlaOp()
|
| /aosp_15_r20/system/bpf/include/vmlinux/android15_6.6/x86_64/ |
| H A D | vmlinux.h | 117750 int (*program_key)(struct ufs_hba *, const union ufs_crypto_cfg_entry *, int); member
|
| /aosp_15_r20/system/bpf/include/vmlinux/android15_6.6/arm64/ |
| H A D | vmlinux.h | 107884 int (*program_key)(struct ufs_hba *, const union ufs_crypto_cfg_entry *, int); member
|
| /aosp_15_r20/kernel/prebuilts/6.1/arm64/ |
| D | abi.stg | 157242 name: "program_key"
|
| D | abi-full.stg | 157354 name: "program_key"
|
| /aosp_15_r20/kernel/prebuilts/6.6/arm64/ |
| D | abi.stg | 162241 name: "program_key"
|
| D | abi-full.stg | 165341 name: "program_key"
|
| /aosp_15_r20/kernel/prebuilts/mainline/arm64/ |
| D | abi-full.stg | 189929 name: "program_key"
|
| D | abi.stg | 189929 name: "program_key"
|