- Sort Score
- Result 10 results
- Languages All
Results 31 - 40 of 134 for constexpr (0.16 sec)
-
tensorflow/compiler/mlir/tf2xla/transforms/tf2xla_rewriter_test.cc
using ::mlir::func::FuncOp; using ::tsl::Status; using ::tsl::StatusOr; using ::xla::ReplicaGroup; using ::xla::ShapeUtil; using ::xla::XlaBuilder; using ::xla::XlaComputation; using ::xla::XlaOp; static constexpr char kMlirModuleStr[] = R"( module attributes {tf.versions = {bad_consumers = [], min_consumer = 0 : i32, producer = 1442 : i32}} {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 09:16:07 UTC 2024 - 11.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/tpu_rewrite_device_util.h
#include "tensorflow/core/util/device_name_utils.h" namespace tensorflow { using tsl::StatusOr; inline constexpr absl::string_view kNumCoresPerReplicaAttr = "num_cores_per_replica"; inline constexpr absl::string_view kTopologyAttr = "topology"; inline constexpr absl::string_view kDeviceAssignmentAttr = "device_assignment"; // A TPU device for execution alongside its associated host CPU device.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Apr 26 09:37:10 UTC 2024 - 11.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tpu_dynamic_layout_pass.cc
#include "tensorflow/core/protobuf/tpu/compile_metadata.pb.h" #include "tensorflow/core/util/device_name_utils.h" namespace mlir { namespace TFTPU { namespace { constexpr char kDeviceAttr[] = "device"; constexpr char kDeviceCPU[] = "CPU"; constexpr char kFuncDeviceAttr[] = "tf.device"; struct TPUDynamicLayoutPass : public TF::PerFunctionAggregateAnalysisConsumerPass<
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 12.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/convert_control_to_data_outputs.cc
using DeviceMap = DenseMap<StringAttr, int64_t>; constexpr int64_t kAnyDevice = 0; constexpr ResourceAndDevice kInvalidResourceAndDevice{kInvalidResourceId, kAnyDevice}; constexpr ResourceAndDevice kUnknownResourceAndDevice{kUnknownResourceId, kAnyDevice}; constexpr char kDeviceAttr[] = "device";
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 28.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/passes/tpu_sharding_identification_pass.cc
using mlir::RankedTensorType; using mlir::StringAttr; using mlir::Value; using mlir::WalkResult; constexpr char kReplicateSharding[] = ""; constexpr char kShardingAttr[] = "mhlo.sharding"; constexpr char kUseSpmdAttr[] = "use_spmd_for_xla_partitioning"; constexpr char kAliasingAttr[] = "tf.aliasing_output"; constexpr char kNumCoresPerReplicaAttr[] = "num_cores_per_replica"; #define GEN_PASS_DEF_TPUSHARDINGIDENTIFICATIONPASS
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 02:01:13 UTC 2024 - 28.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/tpu_rewrite_device_util.cc
// Device coordinates are defined as (x, y, z, core), thus resulting in a rank 4 // topology. constexpr int kTPUTopologyRank = 4; constexpr char kDeviceTPUSystem[] = "TPU_SYSTEM"; constexpr char kDeviceTPU[] = "TPU"; constexpr char kTPUReplicatedCore[] = "TPU_REPLICATED_CORE"; constexpr char kTPUReplicatedHost[] = "TPU_REPLICATED_HOST"; constexpr char kBadIntArrayElementMsg[] = "bad '{0}' attribute at index {1}, not an int";
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Jun 10 20:10:40 UTC 2024 - 32.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/common/quantization_lib/quantization_utils.h
inline constexpr char kDebugModeOpQuantAttrName[] = "debug_quant"; // Used to annotate custom ops if they are quantizable. inline constexpr char kQuantTraitAttrName[] = "_tfl_quant_trait"; enum QuantizationTrait { FullyQuantizable = 0, NotQuantizable = 1 }; inline constexpr absl::string_view QuantTraitValues[] = {"fully_quantizable", "not_quantizable"}; inline constexpr double kNearZeroTolerance = 1.0e-6;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 24 20:30:06 UTC 2024 - 41.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/python/tf_tfl_flatbuffer_helpers.cc
namespace tensorflow { namespace internal { namespace { using ::tensorflow::quantization::PyFunctionLibrary; using ::tflite::optimize::ReducedPrecisionSupport; // Op def string for TFLite_Detection_PostProcess Op. constexpr mlir::StringRef kDetectionPostProcessOp = "name: 'TFLite_Detection_PostProcess' input_arg: { name: " "'raw_outputs/box_encodings' type: DT_FLOAT } input_arg: { name: "
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sun May 12 12:39:37 UTC 2024 - 17.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/experimental/remat/rematerializer_test.cc
AddUse(some_operation(rng), some_tensor(rng)); } } }; }; TEST_F(RematSimulationTest, SimulationAgreesWithReality) { constexpr int kNumOperations = 128; constexpr int kNumTensors = 32; constexpr int kNumUses = kNumOperations * kNumTensors / 4; std::mt19937 rng; for (int i = 0; i < 1024; ++i) { RandomRemat remat(kNumOperations, kNumTensors, kNumUses, rng);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 14 20:57:44 UTC 2023 - 19.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/common/lift_as_function_call.cc
using ::tsl::protobuf::TextFormat; // Default version number for native serialization. constexpr int64_t kDefaultVersion = 9; // Default platform for XlaCallModuleOp. constexpr StringRef kPlatformCpu = "CPU"; // Name of `tf.XlaCallModule`'s dictionary attribute for keeping the // deserialized stablehlo module's attributes. constexpr StringRef kStablehloModuleAttrsAttrName = "_stablehlo_module_attrs";
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 21.8K bytes - Viewed (0)