- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 166 for setFAttr (0.16 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/sparsecore/embedding_sequencing.cc
builder.insert(metadata_op.clone()); for (Operation& op : func.getOps()) { if (!IsTPUOp(&op)) continue; op.setAttr(TF::kReplicationInfoAttr, new_group); } TF::TPUCompilationResultOp new_result = compilation_op.clone(); new_result->setAttr(kTpuCompilationStatus, new_group); builder.insert(new_result); } template <typename OpType> LogicalResult FindAndExcludeOp(func::FuncOp func,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 39.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/sparsecore/embedding_pipelining.cc
/*config_proto=*/builder.getStringAttr(""), /*executor_type=*/builder.getStringAttr("")); caller.setFAttr(symbol); // Set an attribute that our inliner will look for when choosing which // TF::StatefulPartitionedCallOps to inline. if (flag_for_inlining) caller->setAttr(kEmbeddingPipeliningInlineAttr, builder.getBoolAttr(true)); return caller; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 92.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_ops_a_m.cc
if (failed(::mlir::TF::UpdateDataFormat(data_format, this))) return failure(); // Update convolution attributes. (*this)->setAttr("dilations", ShuffleArrayAttr(getDilations(), perm)); (*this)->setAttr("strides", ShuffleArrayAttr(getStrides(), perm)); (*this)->setAttr("explicit_paddings", ShuffleArrayAttr(getExplicitPaddings(), perm, 2)); return success(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 146.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_ops_n_z.cc
mlir::CallInterfaceCallable callee) { // Direct call. if (SymbolRefAttr fAttr = op.getFAttr()) { SymbolRefAttr calleeAttr = callee.get<SymbolRefAttr>(); return op.setFAttr(cast<FlatSymbolRefAttr>(calleeAttr)); } // Indirect call, callee Value is the first operand. return op.setOperand(0, callee.get<Value>()); } void PartitionedCallOp::setCalleeFromCallable(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 09 22:07:10 UTC 2024 - 170.8K bytes - Viewed (0) -
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/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/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/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)