- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 10 for attribute (0.25 sec)
-
tensorflow/c/c_api.cc
op->node.AddAttr(attr_name, attr_val); RecordMutation(graph, *op, "setting attribute"); } void TF_ClearAttr(TF_Graph* graph, TF_Operation* op, const char* attr_name, TF_Status* status) { using tensorflow::RecordMutation; mutex_lock l(graph->mu); op->node.ClearAttr(attr_name); RecordMutation(graph, *op, "clearing attribute"); } void TF_SetFullType(TF_Graph* graph, TF_Operation* op,
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Mon Apr 15 03:35:10 GMT 2024 - 102.3K bytes - Viewed (0) -
tensorflow/c/c_api_test.cc
TF_DeleteTensor(fetchValues[0]); TF_DeleteTensor(fetchValues[1]); } // REGISTER_OP for CApiAttributesTest test cases. // Registers two ops, each with a single attribute called 'v'. // The attribute in one op will have a type 'type', the other // will have list(type). #define ATTR_TEST_REGISTER_OP(type) \ REGISTER_OP("CApiAttributesTestOp" #type) \
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Mon Apr 15 03:35:10 GMT 2024 - 96.9K bytes - Viewed (3) -
tensorflow/c/eager/c_api.cc
if (!s.ok()) { LOG(WARNING) << "Unable to set attribute: " << attr_name; } } void TFE_OpSetAttrInt(TFE_Op* op, const char* attr_name, int64_t value) { auto s = tensorflow::unwrap(op)->SetAttrInt(attr_name, value); if (!s.ok()) { LOG(WARNING) << "Unable to set attribute: " << attr_name; } }
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/experimental/gradients/tape/tape_operation.cc
// TODO(srbs): Manage refcount of ForwardOperation's inputs/outputs. forward_op_.outputs.push_back(retvals[i]); } // TODO(b/166669239): This is needed to support AttrBuilder::Get for string // attributes. Number type attrs and DataType attrs work fine without this. // Consider getting rid of this and making the behavior between number types // and string consistent. forward_op_.attrs.BuildNodeDef();
C++ - Registered: Tue Feb 27 12:39:08 GMT 2024 - Last Modified: Tue Jun 07 01:53:35 GMT 2022 - 9K bytes - Viewed (1) -
tensorflow/c/eager/c_api_test.cc
TFE_OpSetAttrType(var_op, "dtype", TF_INT64); TFE_OpSetAttrShape(var_op, "shape", {}, 0, status); const TFE_OpAttrs* attributes = TFE_OpGetAttrs(var_op); TFE_Op* copy_op = TFE_NewOp(ctx, "VarHandleOp", status); TFE_OpSetAttrType(copy_op, "dtype", TF_FLOAT); TFE_OpAddAttrs(copy_op, attributes); unsigned char is_list = 0; ASSERT_EQ(TF_ATTR_TYPE, TFE_OpGetAttrType(copy_op, "dtype", &is_list, 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/eager/gradients.cc
// TODO(srbs): Manage refcount of ForwardOperation's inputs/outputs. forward_op_->outputs.push_back(retvals[i]); } // TODO(b/166669239): This is needed to support AttrBuilder::Get for string // attributes. Number type attrs and DataType attrs work fine without this. // Consider getting rid of this and making the behavior between number types // and string consistent. forward_op_->attrs.BuildNodeDef();
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Feb 15 09:49:45 GMT 2024 - 19.3K bytes - Viewed (0) -
tensorflow/c/eager/parallel_device/parallel_device_lib.cc
// starting on the next operation. finished_join_.wait(l); } context_ = context; operation_name_ = operation_name; step_id_ = step_id; op_inputs_ = inputs; attributes_ = attributes; expected_max_outputs_ = expected_max_outputs; cancellation_manager_ = &cancellation_manager; execution_state_ = ExecutionState::kReadyToExecute; } start_execute_.notify_one(); }
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/parallel_device/parallel_device.cc
const ParallelDevice& parallel_device, const std::string& parallel_device_name, TFE_Context* context, std::vector<MaybeParallelTensorUnowned> inputs, const char* operation_name, const TFE_OpAttrs* attributes, int expected_max_outputs, TF_Status* status) { absl::optional<std::vector<MaybeParallelTensorOwned>> result; // TODO(allenl): We should remove "TPU" from these op names at the very least,
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) -
tensorflow/c/eager/custom_device_testutil.cc
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); if (TF_GetCode(s) != TF_OK) return;
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) -
tensorflow/c/c_api_function_test.cc
EXPECT_EQ(string("Invalid FunctionDef given to TF_FunctionImportFunctionDef"), string(TF_Message(s_))); } TEST_F(CApiFunctionTest, Attribute) { DefineFunction(func_name_, &func_); // Get non existent attribute TF_Buffer* attr_buf = TF_NewBuffer(); TF_FunctionGetAttrValueProto(func_, "foo_attr", attr_buf, s_); EXPECT_EQ(TF_INVALID_ARGUMENT, TF_GetCode(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)