Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 7 of 7 for function_ref (0.16 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/c/eager/c_api_test.cc

      TFE_Context* ctx = TFE_NewContext(opts, status);
      CHECK_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status);
      TFE_DeleteContextOptions(opts);
    
      string function_def = MatMulFunction();
      TFE_ContextAddFunctionDef(ctx, function_def.data(), function_def.size(),
                                status);
      CHECK_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status);
    
      for (bool clear_cache : {true, false, true}) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Aug 03 20:50:20 UTC 2023
    - 94.6K bytes
    - Viewed (0)
  3. tensorflow/c/c_api_function_test.cc

      }
      return v;
    }
    
    // Specification for an expected edge.
    // src is either:
    // - input name (as it appears in FunctionDef)
    // - name of output tensor (in nested "add:z:0" format)
    // dst is either:
    // - output name (as it appears in FunctionDef)
    // - <name_of_node>:<index_of_this_input_into_node> (this looks the same as
    //      output tensor naming, but it the index is actually an input index)
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Jul 20 22:08:54 UTC 2023
    - 63.6K bytes
    - Viewed (0)
  4. 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)
  5. 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)
  6. 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)
  7. tensorflow/c/c_api.h

    // The return value points to memory that is only usable until the next
    // mutation to *func.
    TF_CAPI_EXPORT extern const char* TF_FunctionName(TF_Function* func);
    
    // Write out a serialized representation of `func` (as a FunctionDef protocol
    // message) to `output_func_def` (allocated by TF_NewBuffer()).
    // `output_func_def`'s underlying buffer will be freed when TF_DeleteBuffer()
    // is called.
    //
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Oct 26 21:08:15 UTC 2023
    - 82.3K bytes
    - Viewed (0)
Back to top