- Sort Score
- Result 10 results
- Languages All
Results 1 - 6 of 6 for custom (0.18 sec)
-
tensorflow/c/experimental/gradients/custom_gradient_test.cc
CHECK_EQ(grad_inputs.size(), 1); grad_inputs[0] = grad_outputs[0]; if (grad_inputs[0]) { grad_inputs[0]->Ref(); } return absl::OkStatus(); } }; // Computes: // // @tf.custom_gradient // def f(input): // def grad(grads): // return grads[0] // return tf.exp(input), grad // outputs = [f(inputs[0])] Status ExpWithPassThroughGrad(AbstractContext* ctx,
C++ - Registered: Tue Mar 26 12:39:09 GMT 2024 - Last Modified: Wed Feb 28 13:53:47 GMT 2024 - 4.8K bytes - Viewed (0) -
tensorflow/c/eager/c_api.cc
#include "tensorflow/core/common_runtime/eager/attr_builder.h" #include "tensorflow/core/common_runtime/eager/context.h" #include "tensorflow/core/common_runtime/eager/custom_device.h" #include "tensorflow/core/common_runtime/eager/custom_device_op_handler.h" #include "tensorflow/core/common_runtime/eager/execute.h" #include "tensorflow/core/common_runtime/eager/placement_utils.h"
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Tue Mar 12 20:00:09 GMT 2024 - 43.9K bytes - Viewed (2) -
tensorflow/c/eager/c_api_experimental.cc
// One of the inputs we're trying to pack is on a custom device. We'll let // the first custom device we see handle all of the packing. auto* custom_device_handle = tensorflow::down_cast<tensorflow::CustomDeviceTensorHandle*>( unwrapped_handle); tensorflow::ImmediateExecutionTensorHandle* result; status->status = custom_device_handle->device()->Pack(
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Apr 11 23:52:39 GMT 2024 - 35.9K bytes - Viewed (3) -
tensorflow/c/eager/parallel_device/parallel_device_lib.cc
for (int component_index = 0; component_index < tensors_.size(); ++component_index) { // TODO(allenl): Add a C API for summarizing tensors. Currently custom // devices limiting themselves to a C API (for ABI compatibility) would need // to implement summarization for component tensors themselves. ImmediateExecutionTensorHandle* component =
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) -
tensorflow/c/eager/c_api_test.cc
TF_DeleteStatus(status); TFE_DeleteOp(var_op); TFE_DeleteOp(var_op_2); TFE_DeleteContext(ctx); } // Needs to work with a const TFE_Op since custom devices should not modify the // op they are called with. TFE_Op* CloneOp(const TFE_Op* other) { TF_Status* status = TF_NewStatus(); TFE_Context* context = TFE_OpGetContext(other, status);
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Aug 03 20:50:20 GMT 2023 - 94.6K bytes - Viewed (1) -
tensorflow/c/c_api_function_test.cc
/*opts=*/nullptr, description, s.get()); ASSERT_EQ(TF_OK, TF_GetCode(s.get())) << TF_Message(s.get()); ASSERT_NE(*func, nullptr); } REGISTER_OP("CustomOp") .Output("output: float32") .Attr("index: int") .SetShapeFn(tensorflow::shape_inference::UnknownShape); void NodeWithPlaceholderAttrHelper(TF_Graph* graph, TF_Status* s,
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Jul 20 22:08:54 GMT 2023 - 63.6K bytes - Viewed (6)