- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 20 for Platen (0.1 sec)
-
tensorflow/c/experimental/saved_model/core/revived_types/partially_revived_objects.cc
// Additionally, we take advantage of the fact that the SignatureDefFunction's // associated functiondef has lexicographically ordered inputs/outputs due to // nest.flatten. Status LoadSignatureDefFunctionMetadata( const SavedConcreteFunction& saved_concrete_function, SignatureDefFunctionMetadata* out) { std::vector<SignatureDefParam> args;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 09 20:11:48 UTC 2023 - 23.7K bytes - Viewed (0) -
tensorflow/cc/gradients/array_grad.cc
auto indices_size = ExpandDims(scope, Size(scope, indices), 0); Output outer_shape, flat_values_shape; if (batch_dims != 0) { auto values_shape = Shape(scope, values); // Add the batch offsets to indices and flatten the batch dimensions. outer_shape = Slice(scope, values_shape, {0}, {batch_dims}); auto inner_shape = Slice(scope, Slice(scope, values_shape, {batch_dims}, {-1}), {1}, {-1});
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/compiler/mlir/lite/stablehlo/transforms/legalize_hlo_conversions/dot_general.cc
rhs, DenseIntElementsAttr::get( RankedTensorType::get({rhs_rank}, rewriter.getI64Type()), rhs_permutation)); // Reshapes lhs to flatten out_dimensions and contracting_dimensions. llvm::SmallVector<int64_t, 4> lhs_flattened_shape = Concat<int64_t>( lhs_dot_dimensions_info.batch_dimensions().SizesArray(), llvm::ArrayRef<int64_t>{
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 19.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/einsum.cc
llvm::StringRef equation, RankedTensorType lhs_ty) { llvm::StringRef lhs; llvm::StringRef out; std::tie(lhs, out) = equation.split("->"); if (lhs.empty() || out.empty()) return std::nullopt; // Try to flatten the "..." if possible. int lhs_named_label, rhs_named_label; // following rhs and rhs_ty variables are non-functional here only created to // comply with the existing API llvm::StringRef rhs;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 33.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/optimize.cc
(std::equal(i1, reduced_e1, i2)); } // Check if the value of the last dimension of type1 is equal to the number of // elements in type2. This is a required condition to flatten type2 to form a // 1D array and allow the binaryOp handle the broadcasting implicitly. bool IsLastDimEqualToNumElements(Type type1, Type type2) { return (mlir::cast<ShapedType>(type1).getRank() >= 1 &&
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 00:40:15 UTC 2024 - 102.3K bytes - Viewed (0) -
tensorflow/cc/experimental/libtf/function.cc
// TODO(b/190203981): Move to a separate nest-like library. void Flatten(const TaggedValue& value, std::vector<AbstractTensorHandle*>* flat_args) { if (value.type() == TaggedValue::Type::TENSOR) { flat_args->emplace_back(value.tensor().get()); } else if (value.type() == TaggedValue::Type::TUPLE) { for (const auto& t : value.tuple()) { Flatten(t, flat_args); } } else {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Mar 04 19:49:06 UTC 2024 - 9.1K bytes - Viewed (0) -
tensorflow/c/experimental/saved_model/core/saved_model_utils.cc
const SavedConcreteFunction& saved_concrete_function, const FunctionDef* function_def) { // tf.functions go through many transformations before becoming FunctionDefs // 1. flatten user-provided inputs: // https://github.com/tensorflow/tensorflow/blob/1c064ab76064c58e54261b805027474885a1534d/tensorflow/python/eager/function.py#L2671-L2675 // 2. convert user-provided inputs to tensors:
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jan 12 19:17:46 UTC 2023 - 24K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/translate/import_model.cc
// linearized function argument or return on a FunctionDef, and hence to an // mlir::func::FuncOp argument / return. // // This must match the linearization that happens in `tf.nest.flatten`. // In particular, dict values should be linearized in sorted key order. // // The linearized index paths can be returned back to a structured // representation (e.g. to emit C structs matching a signature) with a simple
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/lite/tf_tfl_passes.cc
// Add inline pass. pass_manager.addPass(mlir::createInlinerPass()); // Expands mhlo.tuple ops. pass_manager.addPass( mlir::mhlo::createExpandHloTuplesPass(entry_function_name.str())); // Flatten tuples for control flows. pass_manager.addNestedPass<mlir::func::FuncOp>( mlir::mhlo::createFlattenTuplePass()); mlir::odml::AddMhloOptimizationPasses( pass_manager,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 06 18:45:51 UTC 2024 - 25.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_hlo.cc
rhs, DenseIntElementsAttr::get( RankedTensorType::get({rhs_rank}, rewriter.getI64Type()), rhs_permutation)); // Reshapes lhs to flatten out_dimensions and contracting_dimensions. llvm::SmallVector<int64_t, 4> lhs_flattened_shape = Concat<int64_t>( lhs_dot_dimensions_info.batch_dimensions().SizesArray(), llvm::ArrayRef<int64_t>{
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 154.9K bytes - Viewed (0)