Searched defs:execute_op (Results 1 – 9 of 9) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/ |
| H A D | Loops.h | 86 execute_op(char* C10_RESTRICT data[], const int64_t* strides, int64_t i, int64_t n, func_t&& op) { in execute_op() function
101 execute_op(char* C10_RESTRICT data[], const int64_t* strides, int64_t i, int64_t n, func_t&& op) { in execute_op() function
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/transforms/ |
| H A D | remove_device_attribute.cc | 50 module.walk([&](tfrt::corert::ExecuteOp execute_op) { in runOnOperation()
|
| H A D | cross_device_transfer.cc | 80 } else if (auto execute_op = llvm::dyn_cast<tfrt::corert::ExecuteOp>(op)) { in GetDevice() local
|
| H A D | tf_to_tfrt.cc | 1840 if (auto execute_op = in GetNumArgs() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/ |
| H A D | tpu_add_resource_device_attribute.cc | 52 mlir::BlockArgument arg, mlir::TF::TPUExecuteOp execute_op) { in AddPlaceholderDeviceAttributeToResource()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| H A D | tpu_rewrite_pass.cc | 440 OpBuilder* builder, TF::TPUExecuteOp* execute_op) { in BuildExecuteOp()
589 Operation* execute_op, OpBuilder* builder) { in AssignDevicesToReplicatedExecute()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xrt/ |
| H A D | xrt_util.cc | 382 const ChainedExecuteFn& execute_op, in ExecuteChained()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xrt/kernels/ |
| H A D | xrt_execute_op.cc | 562 auto execute_op = [&](const xrt::XRTChainedExecuteOp& op, in DoWork() local
|
| H A D | tpu_execute_op.cc | 430 auto execute_op = [&](const xrt::XRTChainedExecuteOp& op, in DoWork() local
|