Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 4 of 4 for function_ref (0.34 sec)

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

    // that handles the updated value for an resource argument.
    LogicalResult LiftArgRetResourcesForFunction(
        func::FuncOp func_op,
        const llvm::SmallDenseMap<int64_t, Type>& resource_data_types,
        llvm::function_ref<void(int64_t, Value)> handle_updated_arg_value) {
      RegionResourceHoister hoister(func_op);
      if (failed(hoister.Analyze())) return failure();
    
      // Each of these resources could be read or written in the function. If its
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 55.1K bytes
    - Viewed (0)
  2. tensorflow/compiler/jit/encapsulate_subgraphs_pass.cc

      // Find subgraphs marked with 'group_attribute', and build a new
      // subgraph, one for each value of 'group_attribute'.
      Status SplitIntoSubgraphs(FunctionLibraryDefinition* library);
    
      // Build a FunctionDef for each subgraph, and add it 'library'. The values of
      // the 'group_attribute' annotations become the function names.
      // If 'reuse_existing_functions' is set, use an existing function with the
      // same name, if any.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Feb 22 08:47:20 UTC 2024
    - 51K bytes
    - Viewed (0)
  3. tensorflow/compiler/jit/mark_for_compilation_pass_test.cc

    // inside the function can be compiled.
    TEST(XlaCompilationTest, PartitionedCallUnsupported) {
      FunctionDef compilable = FunctionDefHelper::Define(
          "CompilableFn", {"n_a:float", "n_b:float"}, {"n_c:float"}, {},
          {{{"n_c"}, "Add", {"n_a", "n_b"}, {{"T", DT_FLOAT}}}});
      FunctionDef uncompilable =
          FunctionDefHelper::Define("UncompilableFn", {"n_a:float"}, {"n_c:float"},
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Feb 14 10:11:10 UTC 2024
    - 79.6K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/quantization/tensorflow/python/integration_test/quantize_model_test_base.py

        )
    
      def _is_quantized_function(self, func: function_pb2.FunctionDef) -> bool:
        """Determine whether a FunctionDef is quantized.
    
        Args:
          func: A FunctionDef object.
    
        Returns:
          True iff `func` is quantized.
        """
        return func.signature.name.startswith('quantized_')
    
      def _is_composite_function(self, func: function_pb2.FunctionDef) -> bool:
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Mar 21 08:51:46 UTC 2024
    - 51.2K bytes
    - Viewed (0)
Back to top