- Sort Score
- Result 10 results
- Languages All
Results 151 - 160 of 1,170 for const1 (0.08 sec)
-
tensorflow/compiler/mlir/tf2xla/mlir_bridge_rollout_policy.cc
#include <optional> #include "tensorflow/compiler/jit/flags.h" namespace tensorflow { MlirBridgeRolloutPolicy GetMlirBridgeRolloutPolicy( const tensorflow::Graph& graph, const FunctionLibraryDefinition* function_library, std::optional<ConfigProto> config_proto, bool is_supported_by_replicated_brige, bool uses_uninitialized_resource_args, bool is_v1_compat,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Feb 16 23:11:04 UTC 2024 - 1.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/host_runtime/tpu_merge_variables_with_execute.cc
// Remove the assign ops. for (const auto& entry : var_access_info.per_resource_info) { const auto& info = entry.getSecond(); if (info.assign) info.assign->erase(); } // Remove the read ops if they have no more uses. for (const auto& entry : var_access_info.per_resource_info) { const auto& info = entry.getSecond(); if (info.read->use_empty()) info.read->erase();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 29 17:52:11 UTC 2024 - 27K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/attribute_utils.h
// enable logging. inline const char kBridgeComponent[] = "TFXLABridge"; inline const char kMlirPh1BridgeCounterReplicated[] = "replicated"; inline const char kMlirPh1BridgeCounterNonReplicated[] = "nonreplicated"; inline const char kMlirPh1BridgeCounterV1[] = "v1"; inline const char kMlirPh1BridgeCounterV2[] = "v2"; inline const char kMlirPh1BridgeCounterTpu[] = "tpu"; inline const char kMlirPh1BridgeCounterNonTpu[] = "cpu/gpu";
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 22 19:47:48 UTC 2024 - 8.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/common/quantization_lib/quantization_driver.h
explicit QuantizationDriver(func::FuncOp func_op, const bool is_signed, const int bit_width, const bool disable_per_channel, OpQuantSpecGetter op_quant_spec_getter, OpQuantScaleSpecGetter op_quant_scale_spec_getter, const bool infer_tensor_range,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Mar 20 11:42:17 UTC 2024 - 16.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/cc/weight_only_ptq.cc
return module_op; } absl::Status QuantizeWeightOnlyPtq( const absl::string_view src_saved_model_path, const absl::string_view dst_saved_model_path, QuantizationConfig quantization_config, const std::vector<std::string>& signature_keys, const absl::flat_hash_map<std::string, SignatureDef>& signature_def_map, const PyFunctionLibrary& py_function_library) { std::unordered_set<std::string> tags;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 03 02:59:01 UTC 2024 - 5.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/tests/passes/insert_weight_param.mlir
// and function name containing conv. func.func @qdq_for_conv_weight_empty(%arg0: tensor<1x3x2x3xf32>) -> tensor<1x2x2x2xf32> attributes {tf._original_func_name = "main_0"} { %cst = "tf.Const"() {value = dense<3.000000e-01> : tensor<2x3x3x2xf32>} : () -> tensor<2x3x3x2xf32> %0 = "tf.XlaCallModule"(%arg0, %cst) { Sout = [#tf_type.shape<1x2x2x2>], _entry_function = @composite_conv_fn,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 09 05:56:10 UTC 2024 - 22K bytes - Viewed (0) -
tensorflow/cc/framework/grad_op_registry.cc
return grad_op_registry; } bool GradOpRegistry::Register(const string& op, GradFunc func) { CHECK(registry_.insert({op, func}).second) << "Existing gradient for " << op; return true; } Status GradOpRegistry::Lookup(const string& op, GradFunc* func) const { auto iter = registry_.find(op); if (iter == registry_.end()) { const string error_msg = "No gradient defined for op: " + op +
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Apr 13 05:57:22 UTC 2024 - 1.6K bytes - Viewed (0) -
tensorflow/compiler/jit/clone_constants_for_better_clustering_test.cc
// } CloneConstantsForBetterClusteringPass rewriter; TF_RETURN_IF_ERROR(rewriter.Run(options)); *result = std::move(graph); return absl::OkStatus(); } const char* kCPU = "/job:localhost/replica:0/task:0/device:CPU:0"; const char* kGPU = "/job:localhost/replica:0/task:0/device:GPU:0"; TEST(CloneConstantsForBetterClusteringTest, ScalarConstantPlacedOnGpu) { Scope root = Scope::NewRootScope().ExitOnError();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 08:47:20 UTC 2024 - 8.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/mlir_bridge_pass_util_test.cc
TF_ASSERT_OK(root.ToGraph(&graph)); EXPECT_TRUE( IsSupportedByReplicatedBridge(graph, /*function_library=*/nullptr)); } TEST(IsSupportedByReplicatedBridge, ReplicatedModule) { const char* const code = R"mlir( func.func @entry_func_1(%arg0: tensor<i32>) -> tensor<i32> attributes {tf.entry_function = {}} { %0 = "tf.Identity"(%arg0) {_tpu_replicate = "cluster"} : (tensor<i32>) -> (tensor<i32>)
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 17 19:51:50 UTC 2024 - 10.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/metrics/error_collector_inst.cc
inline std::string extract_pass_name(const std::string &signature) { const std::vector<std::string> &v = absl::StrSplit(signature, "::"); return v.back(); } // Errors raised by emitOpError start with "'<dialect>.<op>' op". Returns an // empty string if the pattern is not found or the operator is not in tf or tfl // dialect. inline std::string extract_op_name_from_error_message( const std::string &error_message) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Mar 25 01:48:36 UTC 2024 - 5.3K bytes - Viewed (0)