- Sort Score
- Result 10 results
- Languages All
Results 1 - 7 of 7 for num_inputs (0.45 sec)
-
tensorflow/compiler/jit/extract_outside_compilation_pass.cc
// if they are all fully defined; std::nullopt otherwise. std::optional<std::vector<PartialTensorShape>> GetInferredInputShapes( int num_inputs, Node* send_from_host_node) { std::vector<PartialTensorShape> results(num_inputs); for (int i = 0; i < num_inputs; i++) { const Edge* e; if (!send_from_host_node->input_edge(i, &e).ok()) { return std::nullopt; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 12 06:33:33 UTC 2024 - 104.7K bytes - Viewed (0) -
tensorflow/c/c_api.cc
} void TF_AddInputList(TF_OperationDescription* desc, const TF_Output* inputs, int num_inputs) { std::vector<NodeBuilder::NodeOut> input_list; input_list.reserve(num_inputs); for (int i = 0; i < num_inputs; ++i) { input_list.emplace_back(&inputs[i].oper->node, inputs[i].index); } desc->node_builder.Input(input_list); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 15 03:35:10 UTC 2024 - 102.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_hlo.cc
mlir::FunctionType callee_type = callee.getFunctionType(); SmallVector<Type, 4> expected_callee_input_types; auto num_inputs = op.getInputs().size() / 2; for (unsigned i = 0; i < num_inputs; ++i) { auto input_type = mlir::dyn_cast<RankedTensorType>(op.getOperand(i).getType()); auto scalar = RankedTensorType::get({}, input_type.getElementType());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 154.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/translate/import_model.cc
} } return mlir::UnrankedTensorType::get(element_type); }; // Below we only try and do some shape inference for "source" ops which have // no inputs. if (node.num_inputs() > 0) return default_type(); // Do some simply inference here to get the function arguments correct for // this common case. // TODO(jpienaar): Reconsider post refactoring shape functions.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 01 11:17:36 UTC 2024 - 183.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_ops_n_z.cc
const auto &inputs_ty = op.getInputs().getType(); int n_inputs = inputs_ty.size(); if (n_inputs < 1) return op.emitOpError() << "No inputs"; const auto &init_values_ty = op.getInitValues().getType(); int n_init_values = init_values_ty.size(); if (n_init_values != n_inputs) { return op.emitOpError() << "Number of inputs (" << n_inputs << ") is different than number of init_values ("
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 09 22:07:10 UTC 2024 - 170.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/optimize.cc
int num_input = input_ty.getNumElements(); int input_last_dim = input_ty.getShape().back(); if (input_last_dim < 1) return std::nullopt; int num_result = result_ty.getNumElements(); auto size_last = *(--size.value_end<APInt>()); int32_t k = size_last.getSExtValue(); if (num_input / input_last_dim * k != num_result) return std::nullopt;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 00:40:15 UTC 2024 - 102.3K bytes - Viewed (0) -
src/database/sql/sql_test.go
} }() f() } expectPanic("Exec Exec", func() { db.Exec("PANIC|Exec|WIPE") }) exec(t, db, "WIPE") // check not deadlocked expectPanic("Exec NumInput", func() { db.Exec("PANIC|NumInput|WIPE") }) exec(t, db, "WIPE") // check not deadlocked expectPanic("Exec Close", func() { db.Exec("PANIC|Close|WIPE") }) exec(t, db, "WIPE") // check not deadlocked
Registered: Wed Jun 12 16:32:35 UTC 2024 - Last Modified: Thu May 23 18:42:28 UTC 2024 - 111.6K bytes - Viewed (0)