- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 387 for setAttr (0.39 sec)
-
tensorflow/compat_template.__init__.py
_initializers = _KerasLazyLoader( globals(), submodule="initializers", name="initializers", mode="v2") setattr(_current_module, "losses", _losses) setattr(_current_module, "metrics", _metrics) setattr(_current_module, "optimizers", _optimizers)
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jan 23 02:14:00 UTC 2024 - 2.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/canonicalize_compile_and_replicate_attributes.cc
// Convert `_tpu_replicate`. if (op->hasAttr(TF::kTpuReplicateAttr)) { op->setAttr(tensorflow::kReplicationInfoAttr, op->getAttr(TF::kTpuReplicateAttr)); op->removeAttr(tensorflow::kTpuReplicateAttr); op->setAttr(tensorflow::kCompileDeviceTypeAttr, builder.getStringAttr(tensorflow::kTpuDevice)); } // Convert `_XlaMustCompile`.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Oct 05 23:50:19 UTC 2022 - 4.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/tpu_rewrite_device_util_test.cc
auto cluster = builder.create<mlir::tf_device::ClusterOp>( mlir::UnknownLoc::get(&context), result_types); cluster->setAttr(kNumCoresPerReplicaAttr, builder.getIntegerAttr(builder.getIntegerType(64), 1)); cluster->setAttr(kTopologyAttr, builder.getStringAttr("")); cluster->setAttr(kDeviceAssignmentAttr, builder.getArrayAttr({})); EXPECT_FALSE(HasModelParallelism(cluster)); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Apr 26 09:37:10 UTC 2024 - 46.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tf_device_assignment.cc
if (device_attr.getValue().empty()) { op->setAttr(kDeviceAttr, builder.getStringAttr(default_device_)); } } else if (op->getDialect() == tf) { // Assign default device to all ops in Tensorflow dialect that do not // have device attribute. op->setAttr(kDeviceAttr, builder.getStringAttr(default_device_)); } }); } };
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Oct 05 23:50:19 UTC 2022 - 4.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/experimental/tac/transforms/raise_target_subgraphs.cc
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 11.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/device_util_test.cc
mlir::MLIRContext context; mlir::OwningOpRef<mlir::ModuleOp> module_ref = mlir::ModuleOp::create(mlir::UnknownLoc::get(&context)); mlir::Builder builder(*module_ref); (*module_ref)->setAttr("tf.devices", builder.getBoolAttr(false)); mlir::TF::RuntimeDevices devices; EXPECT_TRUE(mlir::failed(GetDevicesFromOp(*module_ref, &devices))); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 9.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/launch_to_device_attribute.cc
if (op->getDialect() != tf_dialect) return WalkResult::advance(); if (parallel_group_attr) { op->setAttr(TF::kParallelExecAnnotation, parallel_group_attr); } auto device_attr = op->getAttr(kDeviceAttr); if (!device_attr) { op->setAttr(kDeviceAttr, launch.getDeviceAttr()); return WalkResult::advance(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 5.5K bytes - Viewed (0) -
tensorflow/api_template.__init__.py
_initializers = _KerasLazyLoader( globals(), submodule="initializers", name="initializers") setattr(_current_module, "losses", _losses) setattr(_current_module, "metrics", _metrics) setattr(_current_module, "optimizers", _optimizers) setattr(_current_module, "initializers", _initializers) # Do an eager load for Keras' code so that any function/method that needs to
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 05 06:27:59 UTC 2024 - 6.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/common/lift_as_function_call.cc
// function name can be changed by those passes. call_op->setAttr(TF::kStablehloEntryFunctionAttrName, FlatSymbolRefAttr::get(builder.getStringAttr(func_name))); // Store the custom attribute to restore the function name when loading it // back in the post calibration stage. As mentioned above, the above entry // function attribute is not reliable. call_op->setAttr(kOriginalStablehloEntryFunctionAttrName,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 21.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/quantization/lite/tfl_to_std.cc
dq.getArg()); dq.getResult().replaceAllUsesWith(dcast); if (auto extra_attr = op->getAttr(mlir::quant::kVolatileOpAttrName)) { dcast->setAttr(mlir::quant::kVolatileOpAttrName, extra_attr); } dq.erase(); } else if (auto q = llvm::dyn_cast<quantfork::QuantizeCastOp>(op)) { auto out_type = q.getResult().getType();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 22 02:50:01 UTC 2024 - 3.5K bytes - Viewed (0)