- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for reshape_input (0.2 sec)
-
tensorflow/compiler/jit/partially_decluster_pass_test.cc
Output shape = ops::Add(s.WithOpName("shape"), shape_a, shape_b); Output reshape_input = ops::Placeholder(s.WithOpName("reshape_input"), DT_FLOAT, ops::Placeholder::Attrs{}); Output reshape = ops::Reshape(s.WithOpName("reshape"), reshape_input, shape); AddToCluster({shape.node(), reshape.node()}, "cluster_0");
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Jun 10 12:32:39 UTC 2022 - 23K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/experimental/tac/transforms/device_transform_patterns.cc
const int64_t flat_size = input_type.getNumElements(); const int64_t width = flat_size / depth; SmallVector<int64_t, 4> input_new_shape({1, 1, width, depth}); auto reshaped_input = InsertReshapeOp(fc_op.getLoc(), input, input_type.getElementType(), input_new_shape, &rewriter); // Insert a reshape after the weight.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 25.4K bytes - Viewed (0)