Search Options

Results per page
Sort
Preferred Languages
Advance

Results 11 - 20 of 38 for Op (0.16 sec)

  1. tensorflow/c/eager/custom_device_test.cc

      std::unique_ptr<TFE_Op, decltype(&TFE_DeleteOp)> op(
          TFE_NewOp(context.get(), "VarHandleOp", status.get()), TFE_DeleteOp);
      ASSERT_TRUE(TF_GetCode(status.get()) == TF_OK) << TF_Message(status.get());
      TFE_OpSetAttrType(op.get(), "dtype", TF_FLOAT);
      TFE_OpSetAttrShape(op.get(), "shape", {}, 0, status.get());
      TFE_OpSetAttrString(op.get(), "container", "", 0);
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Aug 27 23:39:24 GMT 2020
    - 18.4K bytes
    - Viewed (0)
  2. tensorflow/c/experimental/gradients/not_differentiable.cc

      for (int i = 0; i < grad_inputs.size(); i++) {
        grad_inputs[i] = nullptr;
      }
      return OkStatus();
    }
    
    Status RegisterNotDifferentiable(GradientRegistry* registry, const string& op) {
      return registry->Register(op, [](const ForwardOperation& op) {
        return new NotDifferentiableGradientFunction;
      });
    }
    }  // namespace gradients
    C++
    - Registered: Tue Feb 27 12:39:08 GMT 2024
    - Last Modified: Wed Jun 15 01:15:58 GMT 2022
    - 1.3K bytes
    - Viewed (0)
  3. tensorflow/c/experimental/grappler/grappler.cc

    #include "tensorflow/c/tf_status_helper.h"
    #include "tensorflow/core/framework/function.h"
    #include "tensorflow/core/framework/op.h"
    #include "tensorflow/core/framework/op_def.pb.h"
    #include "tensorflow/core/grappler/costs/graph_properties.h"
    #include "tensorflow/core/grappler/costs/op_performance_data.pb.h"
    #include "tensorflow/core/grappler/grappler_item.h"
    #include "tensorflow/core/grappler/optimizers/custom_graph_optimizer_registry.h"
    C++
    - Registered: Tue Feb 27 12:39:08 GMT 2024
    - Last Modified: Wed Sep 06 19:12:29 GMT 2023
    - 15K bytes
    - Viewed (1)
  4. tensorflow/c/eager/parallel_device/parallel_device_testlib.cc

    }
    
    void Variable::GeneralAssignment(const char* op_name, TFE_Context* context,
                                     TFE_TensorHandle* value, TF_Status* status) {
      std::unique_ptr<TFE_Op, decltype(&TFE_DeleteOp)> op(
          TFE_NewOp(context, op_name, status), &TFE_DeleteOp);
      if (TF_GetCode(status) != TF_OK) return;
      TFE_OpSetAttrType(op.get(), "dtype", type_);
      TFE_OpAddInput(op.get(), handle_, status);
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Tue Jun 15 15:44:44 GMT 2021
    - 12.5K bytes
    - Viewed (0)
  5. tensorflow/c/eager/custom_device_testutil.cc

      if (TF_GetCode(s) != TF_OK) return;
      const char* operation_name = TFE_OpGetName(original_op, s);
      if (TF_GetCode(s) != TF_OK) return;
      const TFE_OpAttrs* attributes = TFE_OpGetAttrs(original_op);
    
      TFE_Op* op(TFE_NewOp(context, operation_name, s));
      if (TF_GetCode(s) != TF_OK) return;
      TFE_OpAddAttrs(op, attributes);
      TFE_OpSetDevice(op, dev->underlying_device.c_str(), s);
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Wed Mar 03 20:47:31 GMT 2021
    - 8.3K bytes
    - Viewed (0)
  6. tensorflow/c/eager/parallel_device/parallel_device_test.cc

      const std::string final_op("Id");
      TFE_OpSetAttrString(op.get(), "final_op", final_op.c_str(),
                          final_op.length());
      TFE_OpSetAttrIntList(op.get(), "subdiv_offsets", nullptr, 0);
    
      TFE_OpAddInput(op.get(), input, status);
      if (TF_GetCode(status) != TF_OK) return nullptr;
    
      TFE_TensorHandle* result_handle;
      int num_retvals = 1;
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Jul 08 23:47:35 GMT 2021
    - 29.3K bytes
    - Viewed (1)
  7. tensorflow/c/eager/gradients_test.cc

      }
    
      AbstractOperationPtr check_numerics_op(ctx->CreateOperation());
      ForwardOperation forward_op;
      Status s = Reset(check_numerics_op.get(), "CheckNumerics",
                       /*raw_device_name=*/nullptr, &forward_op);
      ASSERT_EQ(errors::OK, s.code()) << s.message();
      if (isa<TracingOperation>(check_numerics_op.get())) {
        s = dyn_cast<TracingOperation>(check_numerics_op.get())
                ->SetOpName("check_numerics");
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Feb 15 09:49:45 GMT 2024
    - 7K bytes
    - Viewed (0)
  8. tensorflow/c/eager/parallel_device/parallel_device_lib.cc

            return;
          } else if (execution_state_ == ExecutionState::kReadyToExecute) {
            // op_outputs_ may have been std::moved
            op_outputs_ = std::vector<TensorHandlePtr>();
            Execute(context_, operation_name_, std::move(op_inputs_), attributes_,
                    expected_max_outputs_, &op_outputs_, status_.get());
            execution_state_ = ExecutionState::kHasResult;
          }
        }
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Fri Feb 09 07:47:20 GMT 2024
    - 25.4K bytes
    - Viewed (1)
  9. tensorflow/c/eager/parallel_device/parallel_device.cc

            "first un-packed. Got an un-placed op with an input placed on the "
            "parallel device.");
        return;
      }
      TFE_Context* context = TFE_OpGetContext(original_op, status);
      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 =
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Wed Mar 29 22:05:31 GMT 2023
    - 18.3K bytes
    - Viewed (0)
  10. tensorflow/c/experimental/grappler/grappler_test.cc

      ASSERT_EQ(TF_OK, TF_GetCode(status));
    
      const OpDef* expected_op_def;
      TF_ASSERT_OK(OpRegistry::Global()->LookUpOpDef("Add", &expected_op_def));
      string expected_serialized;
      expected_op_def->SerializeToString(&expected_serialized);
      EXPECT_EQ(expected_serialized, actual_string);
      TF_DeleteBuffer(g_buf);
      TF_DeleteBuffer(op_buf);
      TF_DeleteStatus(status);
      TF_DeleteFunctionLibraryDefinition(func);
    }
    
    C++
    - Registered: Tue Feb 27 12:39:08 GMT 2024
    - Last Modified: Thu Apr 13 22:30:58 GMT 2023
    - 11.6K bytes
    - Viewed (0)
Back to top