Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for TF_AddFunctionParameter (0.15 sec)

  1. tensorflow/c/eager/c_api_unified_experimental_test.cc

      TF_ExecutionContext* graph_ctx = TF_CreateFunction(fn_name.c_str(), s);
      ASSERT_EQ(TF_OK, TF_GetCode(s)) << TF_Message(s);
    
      auto* arg0 = TF_AddFunctionParameter(graph_ctx, TF_FLOAT, {-1, nullptr}, s);
      ASSERT_EQ(TF_OK, TF_GetCode(s)) << TF_Message(s);
      auto* arg1 = TF_AddFunctionParameter(graph_ctx, TF_FLOAT, {-1, nullptr}, s);
      ASSERT_EQ(TF_OK, TF_GetCode(s)) << TF_Message(s);
    
      // Create a first "Add" computing `arg0 + arg1`.
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 39.1K bytes
    - Viewed (0)
  2. tensorflow/c/eager/c_api_unified_experimental.cc

      }
      tsl::Set_TF_Status_from_Status(s,
                                     tracing_ctx->Finalize(unwrap(outputs), &func));
      TF_DeleteExecutionContext(ctx);
      return wrap(func);
    }
    
    TF_AbstractTensor* TF_AddFunctionParameter(TF_ExecutionContext* func,
                                               TF_DataType dtype, TF_Shape shape,
                                               TF_Status* s) {
      DCHECK_GE(shape.num_dims, -1);
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 9K bytes
    - Viewed (0)
Back to top