Home
last modified time | relevance | path

Searched defs:send_op (Results 1 – 7 of 7) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/
H A Drestore_shape_inference.cc93 auto send_op = GetCorrespondingDTensorSendRecvOp<mlir::TF::DTensorRecv>( in BackwardShapeInferenceToRestoreOp() local
H A Dlower_send_recv.cc66 module.walk([&](mlir::TF::DTensorSend send_op) { in LowerDTensorSendRecvsOps()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
H A Drewrite_tpu_embedding_ops.cc66 SendTPUEmbeddingGradientsOp send_op; in RunOnRegion() local
/aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/expansions/
H A Dsave_restore_spmd_expander.cc806 if (auto send_op = in GetLayoutsFromAssignVariableOps() local
/aosp_15_r20/external/mesa3d/src/intel/compiler/
H A Dbrw_eu_emit.c1863 enum opcode send_op, in brw_memory_fence()
/aosp_15_r20/external/tensorflow/tensorflow/core/graph/
H A Dgraph_partition.cc240 const string send_op = (host_memory) ? "_HostSend" : "_Send"; in AddSend() local
/aosp_15_r20/external/mesa3d/src/intel/compiler/elk/
H A Delk_eu_emit.c2869 enum elk_opcode send_op, in elk_memory_fence()