- Sort Score
- Result 10 results
- Languages All
Results 1 - 1 of 1 for kMiniBatchCsrAttr (0.13 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/sparsecore/embedding_program_key.cc
#include "xla/mlir_hlo/mhlo/IR/hlo_ops.h" namespace mlir { namespace TFDevice { namespace { constexpr char kDeviceAttr[] = "device"; constexpr char kMiniBatchSplitsAttr[] = "mini_batch_splits"; constexpr char kMiniBatchCsrAttr[] = "mini_batch_in_csr"; #define GEN_PASS_DEF_EMBEDDINGPROGRAMKEYPASS #include "tensorflow/compiler/mlir/tensorflow/transforms/sparsecore/sparsecore_passes.h.inc" struct EmbeddingProgramKeyPass
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 17.4K bytes - Viewed (0)