- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 57 for num_inputs (0.38 sec)
-
tensorflow/cc/gradients/functional_grad.cc
(*f.mutable_attr())[attr.first] = attr.second; } std::vector<Output> func_inputs; std::vector<DataType> input_dtypes; const int num_inputs = op.num_inputs(); func_inputs.reserve(num_inputs + grad_inputs.size()); input_dtypes.reserve(num_inputs); for (int i = 0; i < num_inputs; i++) { func_inputs.push_back(op.input(i)); input_dtypes.push_back(op.input_type(i)); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Oct 15 20:09:06 UTC 2021 - 2.1K bytes - Viewed (0) -
tensorflow/c/experimental/gradients/grad_test_helper.cc
double abs_error) { auto num_inputs = inputs.size(); std::vector<AbstractTensorHandle*> outputs(num_inputs); auto s = RunModel(grad_model, ctx, inputs, absl::MakeSpan(outputs), /*use_function=*/use_function); ASSERT_EQ(errors::OK, s.code()) << s.message(); for (int i = 0; i < num_inputs; ++i) { if (!outputs[i]) continue;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 28 13:53:47 UTC 2024 - 5K bytes - Viewed (0) -
tensorflow/cc/framework/ops.cc
namespace tensorflow { Operation::Operation(Node* n) : inputs_(GetInputs(n)), node_(n) {} Output Operation::input(int32_t i) const { CHECK_NOTNULL(node_); CHECK_GE(i, 0); CHECK_LT(i, node_->num_inputs()); // Handle the case where the input was unknown at the time this // Operation was constructed. if (inputs_[i].first == nullptr && inputs_[i].second == -1) { for (const Edge* e : node_->in_edges()) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Aug 11 01:20:50 UTC 2021 - 3.5K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_compile_util_test.cc
const Node* identity_node = node_name_index.at("identity_op"); EXPECT_EQ(identity_node->op_def().name(), "Identity"); EXPECT_EQ(identity_node->attrs().FindByString("T")->type(), DT_FLOAT); EXPECT_EQ(identity_node->num_inputs(), 1); const Node* identity_input_node = nullptr; TF_EXPECT_OK(identity_node->input_node(0, &identity_input_node)); EXPECT_EQ(identity_input_node->name(), "_arg0");
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 16 21:48:05 UTC 2023 - 6K bytes - Viewed (0) -
tensorflow/c/c_api_experimental.cc
using tensorflow::shape_inference::ShapeHandle; const int num_inputs = input_shapes->num_items; NodeDef node_def; tensorflow::ImmediateExecutionOperation* op = tensorflow::unwrap(tfe_op); node_def.set_name(op->Name()); node_def.set_op(op->Name()); for (int i = 0; i < num_inputs; ++i) { node_def.add_input("dummy_input"); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 15 03:35:10 UTC 2024 - 29.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/host_runtime/tpu_variable_runtime_reformatting.cc
if (data_type.getIntOrFloatBitWidth() == 64) continue; const auto& block_arg = replicate.GetBody().getArgument(replicate_arg); int64_t num_inputs = 0; if (replicate.IsReplicatedBlockArgument(block_arg)) { num_inputs = num_replicas; } else { num_inputs = 1; } // We have found a mirrored variable which is an input to the replicated
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 21.9K bytes - Viewed (0) -
tensorflow/cc/ops/while_loop_test.cc
namespace tensorflow { namespace { class WhileLoopTest : public ::testing::Test { protected: WhileLoopTest() : scope_(Scope::NewRootScope()) {} void Init(int num_inputs, DataType dtype = DT_INT32) { for (int i = 0; i < num_inputs; ++i) { inputs_.push_back(ops::Placeholder(scope_, dtype)); } } void CreateLoop(const ops::CondGraphBuilderFn& cond,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 13 22:30:58 UTC 2023 - 6.4K bytes - Viewed (0) -
tensorflow/cc/gradients/array_grad.cc
const std::vector<Output>& grad_inputs, std::vector<Output>* grad_outputs) { if (op.num_inputs() < 3) { return errors::InvalidArgument("SplitV requires 3 arguments"); } grad_outputs->push_back(Concat(scope, grad_inputs, op.input(2))); for (int i = 0; i < op.num_inputs() - 1; ++i) { grad_outputs->push_back(NoGradient()); } return scope.status(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Oct 10 23:33:32 UTC 2023 - 31.7K bytes - Viewed (0) -
tensorflow/c/ops.cc
return cc_ctx->num_inputs(); } void TF_ShapeInferenceContextGetInput(TF_ShapeInferenceContext* ctx, int i, TF_ShapeHandle* handle, TF_Status* status) { TF_SetStatus(status, TF_OK, ""); auto* cc_ctx = reinterpret_cast<InferenceContext*>(ctx); if (i < 0 || i >= cc_ctx->num_inputs()) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jul 28 22:41:35 UTC 2022 - 10.9K bytes - Viewed (0) -
tensorflow/compiler/jit/encapsulate_xla_computations_pass.cc
std::vector<const Edge*> in_edges; TF_RETURN_IF_ERROR(launch->input_edges(&in_edges)); const int num_inputs = in_edges.size(); const int variable_start_index = xla_function_info.variable_start_index; const int num_variables = num_inputs - variable_start_index; const int num_args = variable_start_index; VLOG(4) << "Launch node '" << launch->name() << "'"
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 12 06:33:33 UTC 2024 - 15.1K bytes - Viewed (0)