- Sort Score
- Result 10 results
- Languages All
Results 1 - 4 of 4 for permutation (0.14 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/lower_tf.cc
// input_shape[M+1], ..., input_shape[N-1]] SmallVector<int64_t> permutation(reshaped_shape.size()); permutation[0] = block_rank; for (int i = 0; i < block_rank; ++i) { permutation[1 + 2 * i] = block_rank + 1 + i; permutation[1 + 2 * i + 1] = i; } std::iota(permutation.begin() + 1 + block_rank * 2, permutation.end(), 1 + block_rank * 2);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 74.9K bytes - Viewed (0) -
tensorflow/compiler/jit/encapsulate_subgraphs_pass.cc
} } return absl::OkStatus(); } // Renumber the indices of _Arg nodes in a graph, according to // 'permutation' that maps old indices to new indices. static Status RenumberArguments(Graph* graph, const std::vector<int>& permutation) { for (Node* n : graph->op_nodes()) { if (n->type_string() == kArgOp) { int index;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 08:47:20 UTC 2024 - 51K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/optimize_patterns.td
(TFL_TransposeOp (TFL_TransposeOp:$transpose_out1 $input, (Arith_ConstantOp:$permutation1 $p1)), (Arith_ConstantOp:$permutation2 $p2)), (TFL_TransposeOp $input, (Arith_ConstantOp (RemapPermutation $permutation1, $permutation2))), [(HasOneUse $transpose_out1)]>; // Convert expand_dims to reshape if possible. def ConvertExpandDimsToReshape : Pat< (TFL_ExpandDimsOp:$expand_dims_op $input, $dim),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 16 20:31:41 UTC 2024 - 66.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/prepare_tf.cc
// method that it is a 4-D RankedTensorType. Value legalizeFilter(PatternRewriter &rewriter, Location loc, Value filter) const { // Create a constant op for HWIO to OHWI transpose permutation. SmallVector<int, 4> perm = {3, 0, 1, 2}; auto perm_type = tensorflow::GetTypeFromTFTensorShape( {static_cast<int>(perm.size())}, rewriter.getIntegerType(32)); auto perm_attr =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 28 21:49:50 UTC 2024 - 64.6K bytes - Viewed (0)