- Sort Score
- Result 10 results
- Languages All
Results 1 - 7 of 7 for AddInput (0.34 sec)
-
tensorflow/c/c_test_util.cc
TF_AddInput(desc, l); TF_AddInput(desc, r); return TF_FinishOperation(desc, s); } TF_Operation* RandomUniform(TF_Operation* shape, TF_DataType dtype, TF_Graph* graph, TF_Status* s) { TF_OperationDescription* desc = TF_NewOperation(graph, "RandomUniform", "random_uniform"); TF_AddInput(desc, {shape, 0}); TF_SetAttrType(desc, "dtype", dtype);
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Fri Oct 15 03:16:52 GMT 2021 - 17.8K bytes - Viewed (2) -
tensorflow/c/eager/gradients.cc
forward_op_->op_name = op; forward_op_->attrs.Reset(op); return op_->Reset(op, raw_device_name); } Status AddInput(AbstractOperation* op_, AbstractTensorHandle* input, ForwardOperation* forward_op_) { TF_RETURN_IF_ERROR(op_->AddInput(input)); forward_op_->inputs.push_back(input); return absl::OkStatus(); } Status AddInputList(AbstractOperation* op_,
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/c_api_unified_experimental_graph.cc
device_name_ = name; return absl::OkStatus(); } Status AddInput(AbstractTensorHandle* input) override { GraphTensor* t = dyn_cast<GraphTensor>(input); if (!t) { return tensorflow::errors::InvalidArgument( "Unable to cast input to GraphTensor"); } TF_AddInput(op_.get(), t->output_); return absl::OkStatus(); }
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Tue Mar 12 20:00:09 GMT 2024 - 15.4K bytes - Viewed (1) -
tensorflow/c/eager/c_api_experimental_test.cc
ASSERT_TRUE(TF_GetCode(status) == TF_OK) << TF_Message(status); TF_OperationDescription* id_descr = TF_NewOperation(function_graph, "Identity", "id"); TF_SetAttrType(id_descr, "T", TF_INT32); TF_AddInput(id_descr, {arg, 0}); TF_Operation* id = TF_FinishOperation(id_descr, status); ASSERT_TRUE(TF_GetCode(status) == TF_OK) << TF_Message(status); TF_Output input{arg, 0}; TF_Output output{id, 0}; TF_Function* fn =
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Aug 03 03:14:26 GMT 2023 - 31.5K bytes - Viewed (1) -
tensorflow/c/eager/c_api.h
// TFE_DeleteOp() is called before TFE_DeleteContext(). // // Very similar to TF_OperationDescription with some differences: // (1) TF_Output or TFE_TensorHandle* as arguments to TF_AddInput, // TF_AddInputList // (2) TF_ColocateWith, TF_AddControlInput etc. do not make sense. // (3) Implementation detail: Avoid use of NodeBuilder/NodeDefBuilder since
C - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Apr 27 21:07:00 GMT 2023 - 22.8K bytes - Viewed (1) -
tensorflow/c/eager/c_api.cc
return tensorflow::unwrap(op)->DeviceName().c_str(); } void TFE_OpAddInput(TFE_Op* op, TFE_TensorHandle* input, TF_Status* status) { status->status = tensorflow::unwrap(op)->AddInput(tensorflow::unwrap(input)); } void TFE_OpAddInputList(TFE_Op* op, TFE_TensorHandle** inputs, int num_inputs, TF_Status* status) { status->status = tensorflow::unwrap(op)->AddInputList(
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/parallel_device/parallel_device_test.cc
const std::string final_op("Id"); TF_SetAttrString(reduce_desc, "final_op", final_op.c_str(), final_op.length()); TF_SetAttrIntList(reduce_desc, "subdiv_offsets", nullptr, 0); TF_AddInput(reduce_desc, x); TF_Operation* reduce_op = TF_FinishOperation(reduce_desc, status); if (TF_GetCode(status) != TF_OK) return; TF_Operation* operations[]{placeholder_op, reduce_op}; TF_Output y{reduce_op, 0};
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)