- Sort Score
- Result 10 results
- Languages All
Results 151 - 160 of 293 for runOnOperation (0.35 sec)
-
tensorflow/compiler/mlir/quantization/stablehlo/passes/nchw_convolution_to_nhwc.cc
namespace { using ::mlir::stablehlo::ConvDimensionNumbersAttr; class NchwConvolutionToNhwcPass : public impl::NchwConvolutionToNhwcPassBase<NchwConvolutionToNhwcPass> { private: void runOnOperation() override; }; // Rewrites NCHW convolution to NHWC. // * Src dimension numbers: [b, f, 0, 1]x[o, i, 0, 1]->[b, f, 0, 1] // * Dst dimension numbers: [b, 0, 1, f]x[0, 1, i, o]->[b, 0, 1, f]
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 8.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/add_quantization_unit_loc.cc
return "quant-add-quantization-unit-loc"; } StringRef getDescription() const final { return "Add QuantizationUnitLoc to quantizable layers."; } private: void runOnOperation() override; }; // TF graph nodes are imported with one of following location patterns: // FusedLoc[NameLoc(op_type:), ..., NameLoc(node_name@func_name)] or
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 7.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/cluster_formation.cc
#define GEN_PASS_DEF_CLUSTERFORMATIONPASS #include "tensorflow/compiler/mlir/tensorflow/transforms/tf_passes.h.inc" struct ClusterFormationPass : public impl::ClusterFormationPassBase<ClusterFormationPass> { void runOnOperation() override; }; void ReplaceLiveOutExternalUses(llvm::ArrayRef<Value> live_outs, tf_device::LaunchOp launch_op) { Region* launch_op_region = &launch_op.getBody();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Oct 05 13:30:21 UTC 2023 - 6.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/mark_input_output_aliases.cc
#include "tensorflow/compiler/mlir/tensorflow/transforms/tf_passes.h.inc" namespace { struct MarkInputOutputAliasesPass : public impl::MarkInputOutputAliasesPassBase<MarkInputOutputAliasesPass> { void runOnOperation() override; }; constexpr char kAliasingAttr[] = "tf.aliasing_output"; constexpr int kUnassigned = -1; struct AliasInfo { AliasInfo() : input_index(kUnassigned), output_index(kUnassigned) {}
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 05 04:14:26 UTC 2024 - 7.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tpu_cluster_cleanup_attributes.cc
class TPUCleanupClusterAttributesPass : public impl::TPUCleanupClusterAttributesPassBase< TPUCleanupClusterAttributesPass> { public: void runOnOperation() override { auto traverse_op = [&](Operation* op, tf_device::ClusterOp tpu_cluster) { if (isa<tf_device::ClusterOp>(op)) return WalkResult::advance(); op->removeAttr(TF::kReplicationInfoAttr);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Oct 05 23:50:19 UTC 2022 - 3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/replicate_invariant_op_hoisting.cc
#include "tensorflow/compiler/mlir/tensorflow/transforms/tf_passes.h.inc" struct ReplicateInvariantOpHoistingPass : public impl::ReplicateInvariantOpHoistingPassBase< ReplicateInvariantOpHoistingPass> { void runOnOperation() override; }; // Check if op directly uses a key in `virtual_devices`. bool DirectUseOfVirtualDevice(const DictionaryAttr& virtual_devices, Operation* op) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 7.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/op_stat_pass.cc
: accepted_dialects_(accepted_dialects), os_(os), total_ops_(0) {} // Prints the resultant operation statistics pos_t iterating over the module. void runOnOperation() override; // Prints summary of op stats. void PrintSummary(); // Keeps track of dtype counts per op. void CountOp(DenseMap<StringRef, llvm::StringMap<int64_t>> &op_count_map,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 8.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/insert_quantized_functions.cc
} void getDependentDialects(DialectRegistry& registry) const override { registry.insert<TF::TensorFlowDialect, func::FuncDialect>(); } private: void runOnOperation() override; // Returns the function library for the given quantization method and opset // pair. llvm::StringRef GetFunctionLibrary(QuantMethod quantization_method,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Mar 22 05:52:39 UTC 2024 - 8.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/metrics/error_collector_inst_test.cc
registry.insert<TF::TensorFlowDialect>(); } public: MLIR_DEFINE_EXPLICIT_INTERNAL_INLINE_TYPE_ID(MockSuccessPass) explicit MockSuccessPass() = default; private: void runOnOperation() override { getOperation().walk([](Operation* nestedOp) { nestedOp->emitError() << "Error at " << nestedOp->getName().getStringRef().str() << " op"; }); }; };
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Mar 25 01:48:36 UTC 2024 - 8.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/extract_tpu_copy_with_dynamic_shape_op.cc
class ExtractTPUCopyWithDynamicShapeOpPass : public impl::ExtractTPUCopyWithDynamicShapeOpPassBase< ExtractTPUCopyWithDynamicShapeOpPass> { void runOnOperation() override; }; // Finds op that created a given value. If the value is a BlockArgument, this // returns the owner of the Block. Operation* GetOpOfValue(Value value) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 8.6K bytes - Viewed (0)