- Sort Score
- Result 10 results
- Languages All
Results 71 - 80 of 330 for constexpr (0.27 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/replica_id_to_device_ordinal.cc
#include "tensorflow/compiler/mlir/tensorflow/ir/tf_ops.h" #include "tensorflow/compiler/mlir/tensorflow/utils/device_util.h" namespace mlir { namespace TFDevice { namespace { constexpr char kReplicaIdAttr[] = "_xla_replica_id"; constexpr char kDeviceOrdinalAttr[] = "device_ordinal"; #define GEN_PASS_DEF_REPLICAIDTODEVICEORDINALPASS #include "tensorflow/compiler/mlir/tensorflow/transforms/tf_passes.h.inc"
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Oct 05 23:50:19 UTC 2022 - 4K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/experimental/tac/transforms/cost_model.cc
#include "tensorflow/compiler/mlir/lite/ir/tfl_ops.h" namespace mlir { namespace TFL { namespace tac { namespace { // These are just fake costs. constexpr float kDequantCost = 2.0; constexpr float kQuantCost = 2.0; constexpr float kRequantCost = 2.0; // TODO(renjieliu): Ideally this should consider different kinds of SOCs as // well. // Get total bytes transferred.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 7.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/tests/analysis/update_op_cost_in_tfrt_mlir_test.cc
#include "tensorflow/core/platform/test.h" #include "tensorflow/core/tfrt/fallback/cost_recorder.h" #include "tfrt/init_tfrt_dialects.h" // from @tf_runtime namespace tensorflow { namespace { constexpr char kCostAttrName[] = "_tfrt_cost"; constexpr char kOpKeyAttrName[] = "op_key"; absl::flat_hash_map<int64_t, uint64_t> GetOpCostMap(mlir::ModuleOp op) { absl::flat_hash_map<int64_t, uint64_t> op_cost_map; op.walk([&](mlir::Operation* op) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Jul 21 22:52:12 UTC 2023 - 3.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/utils/tf_to_uniform_attribute_utils_test.cc
EXPECT_TRUE(uq_requant_op); return uq_requant_op; } TEST_F(TfToUniformAttributeUtilsTest, UniformQuantizedAddOpAttributes) { TfToUniformAttributeUtilsTestPeer test_peer(&ctx_); constexpr absl::string_view kAddOpExpr = R"mlir( %0 = "tf.Const"() {value = #tf_type<tensor_proto : "0x746674656"> : tensor<1x3x2x2x!tf_type.qint32>} : () -> tensor<1x3x2x2x!tf_type.qint32>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Sep 11 00:47:05 UTC 2023 - 11.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/translate/tf_mlir_translate_registration_test.cc
#include "mlir/Tools/mlir-translate/Translation.h" // from @llvm-project #include "tensorflow/core/platform/test.h" namespace mlir { namespace { class MlirTranslationTest : public ::testing::Test { private: static constexpr char kMlirToGraphFlag[] = "-mlir-to-graph"; public: MlirTranslationTest() : translation_(RegisterTranslation()) { // Create fake command line args so that the parser gets chosen.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jul 20 23:11:32 UTC 2023 - 3.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tpu_resource_partitioning.cc
namespace mlir { namespace TFTPU { namespace { #define GEN_PASS_DEF_TPURESOURCEREADSWRITESPARTITIONINGPASS #include "tensorflow/compiler/mlir/tensorflow/transforms/tf_passes.h.inc" constexpr char kUseSpmdAttr[] = "use_spmd_for_xla_partitioning"; constexpr char kNumCoresPerReplicaAttr[] = "num_cores_per_replica"; struct TPUResourceReadsWritesPartitioningPass : public impl::TPUResourceReadsWritesPartitioningPassBase<
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 06 19:12:29 UTC 2023 - 11.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/quantization/device_target.cc
#include "tensorflow/compiler/mlir/lite/quantization/ir/QuantOps.h" #include "tensorflow/compiler/mlir/lite/quantization/numerical_utils.h" namespace mlir { namespace quant { constexpr int k8Bits = 8; constexpr int k32Bits = 32; constexpr unsigned kSigned = QuantizationFlags::Signed; DeviceTarget::DeviceTarget(MLIRContext* ctx) : ctx_(ctx) { f32_ = FloatType::getF32(ctx_); i8_ = IntegerType::get(ctx_, k8Bits);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Mar 08 10:41:08 UTC 2024 - 7.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/readonly_references_to_resources.cc
#include "tensorflow/compiler/mlir/tensorflow/transforms/passes.h" namespace mlir { namespace TF { namespace { // Location attribute. constexpr StringRef kClassAttr = "_class"; constexpr StringRef kSharedNameAttr = "shared_name"; constexpr StringRef kLocationPrefix = "loc:@"; #define GEN_PASS_DEF_CONVERTREADONLYREFERENCEVARIABLESTORESOURCEVARIABLESPASS
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 7.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/api/v1/compile_tf_graph.cc
{tsl::monitoring::Buckets::Exponential(1, 1.5, 45)}); // There were no MLIR ops so the old bridge was called successfully. constexpr char kOldBridgeNoMlirSuccess[] = "kOldBridgeNoMlirSuccess"; // There were no MLIR ops so the old bridge was called but it failed. constexpr char kOldBridgeNoMlirFailure[] = "kOldBridgeNoMlirFailure"; namespace { // Time the execution of kernels (in CPU cycles). Meant to be used as RAII.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 22:19:26 UTC 2024 - 14K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_tf_xla_call_module_to_stablehlo_pass.cc
#include "tensorflow/compiler/mlir/tensorflow/ir/tf_ops.h" namespace mlir { namespace odml { static constexpr std::string_view kStablehloModuleDefaultEntryFuncName = "main"; static constexpr std::string_view kStablehloFuncNamePrefix = "XlaCallModule"; static constexpr char kShardingAttr[] = "mhlo.sharding"; static constexpr char kShardingName[] = "Sharding"; class RemoveCustomCallWithSharding
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jan 25 09:43:18 UTC 2024 - 10.9K bytes - Viewed (0)