- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 58 for do (0.03 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/tf_device_assignment.cc
// have device attribute. op->setAttr(kDeviceAttr, builder.getStringAttr(default_device_)); } }); } }; #define GEN_PASS_DEF_TFDEVICEASSIGNMENTBYFUNCATTRPASS #include "tensorflow/compiler/mlir/tensorflow/transforms/tf_passes.h.inc" // A pass to perform device assignment for TF dialect ops that do not
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/tensorflow/transforms/decompose_resource_ops_pass.cc
// Note: This module search for cluster ops is a bit wasteful as we could have // collected many cluster ops when we were populating reachable functions. But // we would still need to do a walk to find all clusters that do not // reference any function. for (func::FuncOp func : module.getOps<func::FuncOp>()) { // If we have already applied patterns to a function then we can skip
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Sep 08 20:01:13 UTC 2023 - 8K bytes - Viewed (0) -
tensorflow/c/experimental/ops/resource_variable_ops.cc
See the License for the specific language governing permissions and limitations under the License. ==============================================================================*/ // This file is MACHINE GENERATED! Do not edit. #include "tensorflow/c/experimental/ops/resource_variable_ops.h" #include "tensorflow/c/eager/abstract_context.h" #include "tensorflow/c/eager/abstract_tensor_handle.h"
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 10 19:11:36 UTC 2022 - 5.1K bytes - Viewed (0) -
tensorflow/c/experimental/ops/nn_ops.cc
See the License for the specific language governing permissions and limitations under the License. ==============================================================================*/ // This file is MACHINE GENERATED! Do not edit. #include "tensorflow/c/experimental/ops/nn_ops.h" #include "tensorflow/c/eager/abstract_context.h" #include "tensorflow/c/eager/abstract_tensor_handle.h" #include "tensorflow/c/eager/tracing_utils.h"
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 10 19:11:36 UTC 2022 - 5.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/hoist_replicate_invariant_resource_writes.cc
const auto num_replicas = replicate_op.getN(); auto return_op = llvm::dyn_cast<tf_device::ReturnOp>( replicate_op.getRegion().front().getTerminator()); // Get the new result types. // TODO(prakalps): Do not add a value to returned values if it is already // returned. auto new_result_types = llvm::to_vector<4>(replicate_op->getResultTypes()); for (auto assign : tail_assign_variable_ops) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Nov 03 12:35:38 UTC 2022 - 5.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/device_util.cc
mlir::Builder builder(ctx); // Collect devices with attached metadata. llvm::SmallVector<mlir::NamedAttribute, 8> devices; devices.reserve(device_set->devices().size()); // For device that do not have any metadata, or if we failed to parse metadata // from the DeviceSet, we add a unit attribute to the `tf.devices` attribute. for (Device* device : device_set->devices()) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 6.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/set_tpu_infeed_layout.cc
} auto layout = mlir::GetTPUInfeedLayout( mlir::TupleType::get(builder.getContext(), result_types), builder); if (failed(layout)) return mlir::WalkResult::interrupt(); // Do not append a UnitAttr for the "token" operand here to avoid // compilation failure when exporting the "layouts" attribute to a graph // node. Instead, add the UnitAttr during LegalizeTF pass.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 6.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/experimental/tac/tac_module.cc
if (options_.legalize_to_tflite_ops) { // After we creat the alternative subgraph, we can still do canonicalization // legalization & other optimizations as long as we're not inlining the // function. // And in fact, we probably need to do the proper legalization, for the // compute cost to work. (in case we added some TF ops) pass_manager->addPass(mlir::TFL::CreatePrepareTFPass(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 08 01:19:25 UTC 2023 - 5.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/guarantee_all_funcs_one_use.cc
// pathological cases haven't hit us yet. // // 2. Cloning functions can help by allowing code to be specialized (much as // inlining does). In fact, tf-shape-inference attempts to do specialization // of callees which is difficult if callees have multiple uses. class GuaranteeAllFuncsOneUse : public impl::GuaranteeAllFuncsOneUsePassBase<GuaranteeAllFuncsOneUse> { public:
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Oct 05 23:50:19 UTC 2022 - 4.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/cc/calibration/component.cc
if (absl::Status status = AddCalibrationStatistics( module_op, calibration_data_dir, config.calibration_options(), *py_function_lib_); !status.ok()) { LOG(WARNING) << "Some CustomAggregator ops do not have min or max " "values. Parts of the graph are not quantized. " << status; } return module_op; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 14 06:31:57 UTC 2024 - 9.2K bytes - Viewed (0)