- Sort Score
- Result 10 results
- Languages All
Results 121 - 130 of 293 for runOnOperation (0.18 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/tpu_annotate_dynamic_shape_inputs.cc
#include "tensorflow/compiler/mlir/tensorflow/transforms/tf_passes.h.inc" class TPUAnnotateDynamicShapeInputsPass : public impl::TPUAnnotateDynamicShapeInputsPassBase< TPUAnnotateDynamicShapeInputsPass> { 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 - 6.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/tflite_legalize_hlo.cc
class LegalizeHloToTfLitePass : public impl::LegalizeHloToTfLitePassBase<LegalizeHloToTfLitePass> { public: MLIR_DEFINE_EXPLICIT_INTERNAL_INLINE_TYPE_ID(LegalizeHloToTfLitePass); void runOnOperation() override; }; class ConvertReduceOpToTFLiteArgmax : public ConvertReduceOpToArgMinMax<TFL::ReduceMaxOp, TFL::ArgMaxOp, TFL::ReduceAnyOp, true> { public:
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 6.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/post_quantize.cc
// before feeding them to the model and convert them back to floating point // (i.e. dequantize) as the output. this->emit_quant_adaptor_ops_ = emit_quant_adaptor_ops; } void runOnOperation() override; private: quant::CustomOpMap custom_op_map_; }; // Cleans up unnecessary QDQ pattern for input/output ops. class PostQuantizeRemoveQDQPass
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 17.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tf_saved_model_freeze_variables_test_pass.cc
#include "tensorflow/compiler/mlir/tensorflow/transforms/test_passes.h.inc" struct FreezeVariableTestPass : public impl::FreezeVariablesTestPassBase<FreezeVariableTestPass> { void runOnOperation() override { TF::test_util::FakeSession session; if (tf_saved_model::FreezeVariables(getOperation(), &session).failed()) signalPassFailure(); } }; } // namespace
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Oct 04 09:19:38 UTC 2022 - 1.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/verify_suitable_for_graph_export_pass.cc
#include "tensorflow/compiler/mlir/tensorflow/transforms/tf_passes.h.inc" class VerifySuitableForExportPass : public impl::VerifySuitableForExportPassBase< VerifySuitableForExportPass> { public: void runOnOperation() override { if (failed(tensorflow::VerifyExportSuitable(getOperation()))) return signalPassFailure(); } }; } // namespace
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Oct 05 23:50:19 UTC 2022 - 1.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/split_merged_operands.cc
struct SplitMergedOperandsPass : public impl::SplitMergedOperandsPassBase<SplitMergedOperandsPass> { MLIR_DEFINE_EXPLICIT_INTERNAL_INLINE_TYPE_ID(SplitMergedOperandsPass) void runOnOperation() override; }; LogicalResult DuplicateValueIfNeeded(Operation* op, llvm::DenseSet<Value>* values, OpBuilder* builder) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 24 20:30:06 UTC 2024 - 5.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tpu_parallel_execute_sink_resource_write.cc
// handling for `tf_device.parallel_execute`. struct TPUParallelExecuteSinkResourceWrite : public impl::TPUParallelExecuteSinkResourceWritePassBase< TPUParallelExecuteSinkResourceWrite> { void runOnOperation() override; }; // Finds an AssignVariableOp that can be moved into the parallel_execute region. // These AssignVariableOps must be the only consumer of the respective
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Dec 06 04:46:18 UTC 2022 - 6.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/partitioned_topological_sort.cc
const std::function<bool(Operation *)> &partition = IsFlexDelegate) : partition_(partition) {} void runOnOperation() override; private: const std::function<bool(Operation *)> partition_; }; void PartitionedTopologicalSortPass::runOnOperation() { func::FuncOp func = getOperation(); if (!PartitionedTopologicalSort(func, &func.getBody().front(), partition_)) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Dec 19 15:05:28 UTC 2023 - 7.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tpu_update_embedding_enqueue_op_inputs.cc
struct TPUUpdateEmbeddingEnqueueOpInputsPass : public impl::TPUUpdateEmbeddingEnqueueOpInputsPassBase< TPUUpdateEmbeddingEnqueueOpInputsPass> { void runOnOperation() override; }; // Extracts `_tpu_embedding_layer` attribute from TPU embedding ops and // clear the attribute from the operation. This ensures that future optimization
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Oct 05 23:50:19 UTC 2022 - 7.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/readonly_references_to_resources.cc
class ConvertReadonlyReferenceVariablesToResourceVariablesPass : public impl::ConvertReadonlyReferenceVariablesToResourceVariablesPassBase< ConvertReadonlyReferenceVariablesToResourceVariablesPass> { void runOnOperation() override; }; // Parse node name from "_class" or "shared_name" attributes. StringRef GetNodeNameFromClassAttrOrSharedNameAttr(Operation *op) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 7.8K bytes - Viewed (0)