Home
last modified time | relevance | path

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

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