- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for TPUUpdateEmbeddingEnqueueOpInputsPass (0.35 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/tpu_update_embedding_enqueue_op_inputs.cc
#define GEN_PASS_DEF_TPUUPDATEEMBEDDINGENQUEUEOPINPUTSPASS #include "tensorflow/compiler/mlir/tensorflow/transforms/tf_passes.h.inc" struct TPUUpdateEmbeddingEnqueueOpInputsPass : public impl::TPUUpdateEmbeddingEnqueueOpInputsPassBase< TPUUpdateEmbeddingEnqueueOpInputsPass> { void runOnOperation() override; }; // Extracts `_tpu_embedding_layer` attribute from TPU embedding ops and
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Oct 05 23:50:19 UTC 2022 - 7.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tf_passes.td
at the head of TPU computation if it's only used by outside compiled ops. }]; let constructor = "TFTPU::CreateTPUHostComputationExpansionPass()"; } def TPUUpdateEmbeddingEnqueueOpInputsPass : Pass<"tf-tpu-update-embedding-enqueue-op-inputs", "mlir::func::FuncOp"> { let summary = "Updates inputs to TPU embedding enqueue ops depending on whether graph is in training mode or in evaluation mode.";
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 21:18:05 UTC 2024 - 99.6K bytes - Viewed (0)