Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 3 of 3 for TFE_OpGetAttrs (0.11 sec)

  1. tensorflow/c/eager/parallel_device/parallel_device_lib_test.cc

      ASSERT_TRUE(TF_GetCode(status.get()) == TF_OK) << TF_Message(status.get());
      auto outputs =
          parallel_device.Execute(context.get(), std::vector<ParallelTensor*>(),
                                  "VarHandleOp", TFE_OpGetAttrs(handle_op.get()),
                                  /*expected_max_outputs=*/1, status.get());
      ASSERT_TRUE(TF_GetCode(status.get()) == TF_OK) << TF_Message(status.get());
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Mon Oct 21 04:14:14 UTC 2024
    - 15.6K bytes
    - Viewed (0)
  2. tensorflow/c/eager/parallel_device/parallel_device.cc

      if (TF_GetCode(status) != TF_OK) return;
      const char* operation_name = TFE_OpGetName(original_op, status);
      if (TF_GetCode(status) != TF_OK) return;
      const TFE_OpAttrs* attributes = TFE_OpGetAttrs(original_op);
    
      NamedParallelDevice* named_device =
          reinterpret_cast<NamedParallelDevice*>(device_info);
      std::vector<MaybeParallelTensorUnowned> typed_inputs;
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Mon Oct 21 04:14:14 UTC 2024
    - 18.3K bytes
    - Viewed (0)
  3. tensorflow/c/eager/c_api.cc

    void TFE_ContextStartStep(TFE_Context* ctx) {
      tensorflow::unwrap(ctx)->StartStep();
    }
    
    void TFE_ContextEndStep(TFE_Context* ctx) {
      tensorflow::unwrap(ctx)->EndStep();
    }
    
    const TFE_OpAttrs* TFE_OpGetAttrs(const TFE_Op* op) {
      return tensorflow::wrap(tensorflow::unwrap(op)->GetOpAttrs());
    }
    
    void TFE_OpAddAttrs(TFE_Op* op, const TFE_OpAttrs* attrs) {
      tensorflow::unwrap(op)->AddAttrs(tensorflow::unwrap(attrs));
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 43.9K bytes
    - Viewed (0)
Back to top