- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 24 for add_input (0.11 sec)
-
tensorflow/c/kernels/summary_op_test.cc
def.set_device(DEVICE_CPU); AttrValue valuesTypeAttr; SetAttrValue(values->dtype(), &valuesTypeAttr); (*def.mutable_attr())["T"] = valuesTypeAttr; def.add_input(strings::StrCat("input1: ", DataTypeString(tags->dtype()))); def.add_input(strings::StrCat("input2: ", DataTypeString(values->dtype()))); std::unique_ptr<OpKernel> kernel = CreateOpKernel(DeviceType(DEVICE_CPU), nullptr, nullptr, def, 1, &status);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Jul 18 15:10:51 UTC 2022 - 6.7K bytes - Viewed (0) -
tensorflow/c/kernels/bitcast_op_test.cc
SetAttrValue(input_tensor->dtype(), &typeAttr); AttrValue outTypeAttr; SetAttrValue(out_type, &outTypeAttr); (*def.mutable_attr())["T"] = typeAttr; (*def.mutable_attr())["type"] = outTypeAttr; def.add_input( strings::StrCat("input1: ", DataTypeString(input_tensor->dtype()))); std::unique_ptr<OpKernel> kernel = CreateOpKernel(DeviceType(DEVICE_CPU), nullptr, nullptr, def, 1, &status);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Jul 18 15:10:51 UTC 2022 - 5.5K bytes - Viewed (0) -
tensorflow/compiler/jit/tests/device_compiler_test_helper.cc
const std::pair<std::string, FunctionDefHelper::AttrValueWrapper>> attrs) { NodeDef node; node.set_name(std::string(name)); node.set_op(std::string(op)); for (const auto& input : inputs) node.add_input(input); for (const auto& attr : attrs) node.mutable_attr()->insert({attr.first, attr.second.proto}); return node; } } // namespace GraphDef DeviceCompilerSerializeTest::GetTestGraph(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Feb 09 08:24:16 UTC 2024 - 6.2K bytes - Viewed (0) -
tensorflow/c/experimental/ops/gen/cpp/golden/testing_ops.cc.golden
TF_RETURN_IF_ERROR(op_ptr->Reset("AccumulatorApplyGradient", raw_device_name)); TF_RETURN_IF_ERROR(MaybeSetOpName(op_ptr.get(), name)); TF_RETURN_IF_ERROR(op_ptr->AddInput(handle)); TF_RETURN_IF_ERROR(op_ptr->AddInput(local_step)); TF_RETURN_IF_ERROR(op_ptr->AddInput(gradient)); int num_retvals = 0; std::vector<AbstractTensorHandle*> dummy_outputs; return op_ptr->Execute(absl::MakeSpan(dummy_outputs), &num_retvals); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Nov 16 19:04:03 UTC 2023 - 6.5K bytes - Viewed (0) -
tensorflow/c/experimental/ops/nn_ops.cc
TF_RETURN_IF_ERROR( op_ptr->Reset("SparseSoftmaxCrossEntropyWithLogits", raw_device_name)); TF_RETURN_IF_ERROR(MaybeSetOpName(op_ptr.get(), name)); TF_RETURN_IF_ERROR(op_ptr->AddInput(features)); TF_RETURN_IF_ERROR(op_ptr->AddInput(labels)); int num_retvals = 2; AbstractTensorHandle* temp_outputs[2]; Status status = op_ptr->Execute(temp_outputs, &num_retvals); *loss = temp_outputs[0]; *backprop = temp_outputs[1];
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 10 19:11:36 UTC 2022 - 5.9K bytes - Viewed (0) -
tensorflow/c/experimental/ops/io_ops.cc
TF_RETURN_IF_ERROR(op_ptr->Reset("RestoreV2", raw_device_name)); TF_RETURN_IF_ERROR(MaybeSetOpName(op_ptr.get(), name)); TF_RETURN_IF_ERROR(op_ptr->AddInput(prefix)); TF_RETURN_IF_ERROR(op_ptr->AddInput(tensor_names)); TF_RETURN_IF_ERROR(op_ptr->AddInput(shape_and_slices)); TF_RETURN_IF_ERROR( op_ptr->SetAttrTypeList("dtypes", dtypes.data(), dtypes.length())); int num_retvals = tensors.size();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 17 17:54:34 UTC 2022 - 3.9K bytes - Viewed (0) -
tensorflow/c/experimental/saved_model/core/ops/variable_ops.cc
TF_RETURN_IF_ERROR(assign_op->Reset("AssignVariableOp", nullptr)); TF_RETURN_IF_ERROR(assign_op->SetAttrType("dtype", dtype)); TF_RETURN_IF_ERROR(assign_op->AddInput(variable_handle)); TF_RETURN_IF_ERROR(assign_op->AddInput(value)); int num_retvals = 0; TF_RETURN_IF_ERROR(assign_op->Execute({}, &num_retvals)); return Status(); } Status ReadVariable(ImmediateExecutionContext* ctx,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 23 11:28:19 UTC 2024 - 5K bytes - Viewed (0) -
tensorflow/c/experimental/ops/array_ops.cc
TF_RETURN_IF_ERROR(op_ptr->Reset("ExpandDims", raw_device_name)); TF_RETURN_IF_ERROR(MaybeSetOpName(op_ptr.get(), name)); TF_RETURN_IF_ERROR(op_ptr->AddInput(input)); TF_RETURN_IF_ERROR(op_ptr->AddInput(dim)); int num_retvals = 1; return op_ptr->Execute(absl::MakeSpan(output, 1), &num_retvals); } // Op: OnesLike()
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 10 19:11:36 UTC 2022 - 6.7K bytes - Viewed (0) -
tensorflow/c/experimental/ops/resource_variable_ops.cc
TF_RETURN_IF_ERROR(op_ptr->Reset("AssignVariableOp", raw_device_name)); TF_RETURN_IF_ERROR(MaybeSetOpName(op_ptr.get(), name)); TF_RETURN_IF_ERROR(op_ptr->AddInput(resource)); TF_RETURN_IF_ERROR(op_ptr->AddInput(value)); TF_RETURN_IF_ERROR(op_ptr->SetAttrBool("validate_shape", validate_shape)); int num_retvals = 0; std::vector<AbstractTensorHandle*> dummy_outputs;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 10 19:11:36 UTC 2022 - 5.1K bytes - Viewed (0) -
tensorflow/c/experimental/saved_model/core/ops/restore_ops.cc
TF_RETURN_IF_ERROR( CreateStringVectorTensorHandle(ctx, "", &shapes_and_slices_handle)); TF_RETURN_IF_ERROR(restore_op->AddInput(prefix_handle.get())); TF_RETURN_IF_ERROR(restore_op->AddInput(names_handle.get())); TF_RETURN_IF_ERROR(restore_op->AddInput(shapes_and_slices_handle.get())); AbstractTensorHandle* restored_handle = nullptr; int num_retvals = 1; TF_RETURN_IF_ERROR(restore_op->Execute(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Aug 11 01:20:50 UTC 2021 - 4.5K bytes - Viewed (0)