- Sort Score
- Result 10 results
- Languages All
Results 1 - 9 of 9 for input_arg_size (0.27 sec)
-
tensorflow/compiler/jit/get_compiler_ir.cc
// `input_args` includes both concrete_fn input args and captured_input here. auto& input_args = fbody->record->fdef().signature().input_arg(); // Here input_arg_size = len(flat_args) + len(captured_input) int input_arg_size = input_args.size(); std::vector<const Tensor*> inputs(input_arg_size); std::deque<Tensor> inputs_storage; std::vector<VariableInfo> variable_infos; int offset = input_arg_shape_and_dtype.size();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 06:59:07 UTC 2024 - 19K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/c/c_api_unified_experimental_mlir.cc
Builder builder(context_); if (current_ods_input_ != op_def_->input_arg_size()) return InvalidArgument(absl::StrCat("Mismatch in operands number: got ", current_ods_input_, " expected ", op_def_->input_arg_size(), " ; for op ", state_->name.getStringRef().str()));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 28.2K bytes - Viewed (0) -
tensorflow/cc/framework/fuzzing/cc_op_fuzz_gen.cc
std::cout << "NOT fuzzing: " << op_info.graph_op_def.name() << " has a constrained kernel.\n"; return false; } for (int i = 0; i < op_info.graph_op_def.input_arg_size(); ++i) { const auto& arg(op_info.graph_op_def.input_arg(i)); // TODO(unda, b/249298521): deal with inputs that are required to be refs if (arg.is_ref()) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Jan 27 16:26:51 UTC 2024 - 13K bytes - Viewed (0) -
tensorflow/c/experimental/saved_model/core/saved_model_utils.cc
if (input_specs.size() + saved_concrete_function.bound_inputs_size() != function_def->signature().input_arg_size()) { return errors::FailedPrecondition( "FunctionDef ", name, " has ", function_def->signature().input_arg_size(), " inputs, but the SavedConcreteFunction has ", input_specs.size(), " flattened user inputs and ",
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jan 12 19:17:46 UTC 2023 - 24K bytes - Viewed (0) -
tensorflow/c/ops_test.cc
::tensorflow::OpList op_list; op_list.ParseFromArray(op_list_buffer->data, op_list_buffer->length); bool found = false; for (const auto& op : op_list.op()) { if (op.name() == "SomeOp") { ASSERT_EQ(2, op.input_arg_size()); ASSERT_EQ("input1", op.input_arg(0).name()); ASSERT_EQ(::tensorflow::DT_UINT8, op.input_arg(0).type()); ASSERT_EQ(1, op.attr_size()); ASSERT_EQ("string", op.attr(0).type()); found = true;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Aug 11 01:20:50 UTC 2021 - 12.6K bytes - Viewed (0) -
tensorflow/compiler/jit/rearrange_function_argument_pass_test.cc
// and output types should be {DT_BOOL}. const FunctionDef *f1_rewritten = fld.Find("f1_rearrange_0"); CHECK_NE(f1_rewritten, nullptr); ASSERT_EQ(f1_rewritten->signature().input_arg_size(), 2); EXPECT_EQ(f1_rewritten->signature().input_arg(0).type(), DT_BOOL); EXPECT_EQ(f1_rewritten->signature().input_arg(1).type(), DT_RESOURCE); ASSERT_EQ(f1_rewritten->signature().output_arg_size(), 1);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Feb 09 11:36:41 UTC 2024 - 10.5K bytes - Viewed (0) -
tensorflow/cc/framework/cc_op_gen.cc
string body; string return_on_error = strings::StrCat("if (!", scope_str, ".ok()) return;"); strings::StrAppend(&body, " ", return_on_error, "\n"); for (int i = 0; i < op_info.graph_op_def.input_arg_size(); ++i) { const auto& arg(op_info.graph_op_def.input_arg(i)); const auto& api_def_arg(op_info.api_def.in_arg(i)); strings::StrAppend(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 27 17:22:47 UTC 2023 - 17K bytes - Viewed (0) -
tensorflow/cc/framework/cc_op_gen_util.cc
} } void InferOpAttributes( const OpDef& op_def, std::unordered_map<string, string>* inferred_input_attrs) { for (int i = 0; i < op_def.input_arg_size(); ++i) { const auto& arg(op_def.input_arg(i)); InferArgAttributes(arg, inferred_input_attrs); } } bool ArgIsList(const OpDef::ArgDef& arg) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Feb 26 00:57:05 UTC 2024 - 25K bytes - Viewed (0) -
tensorflow/c/c_api_experimental.cc
const tensorflow::OpDef* op_def = nullptr; status->status = tensorflow::OpRegistry::Global()->LookUpOpDef(op_name, &op_def); if (!status->status.ok()) return nullptr; if (input_index >= op_def->input_arg_size() || input_index < 0) { status->status = tensorflow::errors::InvalidArgument( input_index, " out of range for ", op_name); return nullptr; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 15 03:35:10 UTC 2024 - 29.4K bytes - Viewed (0)