- Sort Score
- Result 10 results
- Languages All
Results 1 - 1 of 1 for UpdateEmbeddingEnqueueOpInput (0.38 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/tpu_update_embedding_enqueue_op_inputs.cc
// If SendTPUEmbeddingGradients op is present, this means that graph is in // training mode. As so, correctly feed in `then` branch value of SelectV2 // operand as inputs to the TPU embedding enqueue ops. LogicalResult UpdateEmbeddingEnqueueOpInput( const llvm::StringMap<Operation*>& enqueue_op_map, const llvm::StringMap<Operation*>& recv_activation_op_map, const llvm::StringMap<Operation*>& send_gradient_op_map, OpBuilder* builder) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Oct 05 23:50:19 UTC 2022 - 7.7K bytes - Viewed (0)