- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for mutable_tensor_shape (0.32 sec)
-
tensorflow/compiler/jit/encapsulate_subgraphs_pass_test.cc
NodeBuilder node_builder(opts.GetNameForOp("Const"), "Const", opts.op_registry()); TensorProto value; value.set_dtype(dtype); for (int dim : shape) { value.mutable_tensor_shape()->add_dim()->set_size(dim); } return opts.WithAttr("value", value) .WithAttr("dtype", dtype) .FinalizeBuilder(&node_builder); } Node* KnownShape(absl::Span<const int> shape,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 08:47:20 UTC 2024 - 113.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/translate/import_model.cc
inputs.reserve(assets.size()); for (const auto& asset : assets) { TensorInfo tensor_info; tensor_info.set_name(asset.tensor_name); tensor_info.set_dtype(DT_STRING); tensor_info.mutable_tensor_shape(); inputs.push_back({asset.tensor_name, tensor_info}); } std::unordered_map<std::string, std::string> tf_name_to_mlir_name; TF_ASSIGN_OR_RETURN(auto sub_module,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 01 11:17:36 UTC 2024 - 183.2K bytes - Viewed (0)