Search Options

Results per page
Sort
Preferred Languages
Advance

Results 11 - 20 of 230 for GetOperation (0.43 sec)

  1. tensorflow/compiler/mlir/tensorflow/transforms/remove_unused_arguments.cc

        if (!op.isPrivate()) return;
    
        auto call = llvm::dyn_cast<CallableOpInterface>(op.getOperation());
        if (!call) return;
        Region* region = call.getCallableRegion();
        if (!region) return;  // happens e.g. for external functions
    
        auto func = llvm::dyn_cast<FunctionOpInterface>(op.getOperation());
        if (!func || do_not_touch.count(func)) return;
        llvm::BitVector unused_args(func.getNumArguments());
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 8.9K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tensorflow/ir/tf_ops_layout_helper.h

      // Update types for all layout sensitive results.
      auto layout_sensitive = cast<LayoutSensitiveInterface>(op->getOperation());
      for (unsigned idx : layout_sensitive.GetLayoutDependentResults()) {
        OpResult result = op->getOperation()->getResult(idx);
        result.setType(ShuffleRankedTensorType(result.getType(), perm));
      }
    
      return success();
    }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Jun 08 01:19:25 UTC 2023
    - 5.3K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tensorflow/transforms/executor_tpuv1_outline_tpu_island.cc

      func::FuncOp callee = from.lookup<func::FuncOp>(symbol_ref.getValue());
      callee.getOperation()->getBlock()->getOperations().remove(
          callee.getOperation());
      to.insert(callee);
    }
    
    void TPUBridgeExecutorIslandOutlining::runOnOperation() {
      MLIRContext *ctx = &getContext();
    
      SymbolTable symbol_table(getOperation());
      if (Operation *nested_module = symbol_table.lookup(kNestedModule)) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 8.7K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/tensorflow/transforms/device_index_selector.cc

        : public impl::DeviceIndexSelectorPassBase<DeviceIndexSelector> {
      void runOnOperation() override;
    };
    
    }  // namespace
    
    void DeviceIndexSelector::runOnOperation() {
      func::FuncOp func = getOperation();
      // Convert all the DeviceIndex ops to constant values.
      func.getBody().walk([](TF::DeviceIndexOp op) {
        // This just selects the default in all cases where DeviceIndex feeds into
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Nov 03 12:35:38 UTC 2022
    - 3.4K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/tensorflow/transforms/tpu_parallel_execute_sink_resource_write.cc

      if (resource_handle_op == parallel_execute) return nullptr;
    
      if (resource_handle_op &&
          resource_handle_op->getBlock() ==
              parallel_execute.getOperation()->getBlock() &&
          parallel_execute.getOperation()->isBeforeInBlock(resource_handle_op))
        return nullptr;
    
      return assign_var;
    }
    
    // Finds AssignVariableOps that can be moved into a parallel_execute region and
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Dec 06 04:46:18 UTC 2022
    - 6.6K bytes
    - Viewed (0)
  6. staging/src/k8s.io/apiserver/pkg/admission/plugin/namespace/lifecycle/admission.go

    		// we are not looking at stale state.
    		if a.GetOperation() == admission.Delete {
    			l.forceLiveLookupCache.Add(a.GetName(), true, forceLiveLookupTTL)
    		}
    		// allow all operations to namespaces
    		return nil
    	}
    
    	// always allow deletion of other resources
    	if a.GetOperation() == admission.Delete {
    		return nil
    	}
    
    Registered: Sat Jun 15 01:39:40 UTC 2024
    - Last Modified: Wed Sep 15 09:52:18 UTC 2021
    - 8.6K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/tensorflow/transforms/sink_constant.cc

          llvm::function_ref<bool(tf_device::ClusterOp, ElementsAttr)> filter)
          : filter_(filter) {}
    
      void runOnOperation() override {
        getOperation().walk([filter = filter_](tf_device::ClusterOp cluster) {
          LLVM_DEBUG(llvm::dbgs() << "Visit " << *cluster.getOperation() << "\n");
          // For each launch op, we find the values used that come from a constant
          // defined above and sink these constants in the region body.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Nov 03 12:35:38 UTC 2022
    - 4.3K bytes
    - Viewed (0)
  8. tensorflow/compiler/mlir/tensorflow/transforms/init_text_file_to_import_test_pass.cc

        return "generate a temporary file and invoke InitTextFileToImportPass";
      }
    
     private:
      void runOnOperation() override;
    };
    
    void InitTextFileToImportTestPass::runOnOperation() {
      ModuleOp module = getOperation();
    
      // Create a temporary vocab file.
      int fd;
      SmallString<256> filename;
      std::error_code error_code =
          llvm::sys::fs::createTemporaryFile("text", "vocab", fd, filename);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Oct 04 09:19:38 UTC 2022
    - 5.5K bytes
    - Viewed (0)
  9. tensorflow/compiler/mlir/tensorflow/transforms/replicate_tensor_list_init_ops_pass.cc

      std::vector<OpOperand*> uses;
      for (auto& use : tensor_list.getUses()) {
        uses.emplace_back(&use);
      }
      OpBuilder builder(tensor_list_op.getOperation());
      for (OpOperand* operand : uses) {
        auto new_op = builder.clone(*tensor_list_op.getOperation());
        operand->set(new_op->getResult(0));
      }
    }
    
    // This transformation pass replicates TensorList initialization ops.
    class ReplicateTensorListInitOps
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Sun Jan 22 17:28:34 UTC 2023
    - 2.7K bytes
    - Viewed (0)
  10. tensorflow/compiler/mlir/tensorflow/transforms/lift_variables_test_pass.cc

     public:
      LiftVariablesTestPass() { session_ = new TF::test_util::FakeSession(); }
    
      ~LiftVariablesTestPass() override { delete session_; }
    
      void runOnOperation() override {
        ModuleOp module = getOperation();
        if (failed(tf_saved_model::LiftVariables(module, session_)))
          signalPassFailure();
      }
    
     private:
      Session* session_;
    };
    
    #define GEN_PASS_DEF_LIFTVARIABLESINVALIDSESSIONTESTPASS
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Oct 04 09:19:38 UTC 2022
    - 2.7K bytes
    - Viewed (0)
Back to top