- Sort Score
- Result 10 results
- Languages All
Results 31 - 40 of 111 for _output_shapes (0.28 sec)
-
tensorflow/compiler/mlir/tensorflow/tests/name_anonymous_iterators.mlir
func.func private @gives_a_name_to_anonymous_iterators() { // CHECK: "tf.Iterator" // CHECK-SAME: output_shapes{{.*}}200x28x28x1{{.*}}200x10 // CHECK-SAME: output_types = [f32, f32] // CHECK-SAME: shared_name = "_iterator1" %0 = "tf.AnonymousIteratorV3"() {output_shapes = [ #tf_type.shape<200x28x28x1>, #tf_type.shape<200x10>], output_types = [f32, f32]} : () -> tensor<!tf_type.resource>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Oct 14 09:25:38 UTC 2022 - 1.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/tests/tf_to_corert/func_attributes_multiple_callers.mlir
%3 = "tf.RangeDataset"(%0, %1, %2) {device = "/device:CPU:0", output_shapes = [#tf_type.shape<>], output_types = [i64], metadata = ""} : (tensor<i64>, tensor<i64>, tensor<i64>) -> tensor<!tf_type.variant> // CHECK: tfrt_fallback_async.executeop key({{[0-9]+}}) cost({{.*}}) device("/device:CPU:0") "tf.FlatMapDataset"({{.*}}) {Targuments = [], metadata = "", output_shapes = [#corert.shape<>], output_types = [i64]} {f = "funcB"} : 1
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Nov 17 20:57:36 UTC 2022 - 4.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/tests/tf_to_corert/func_attributes.mlir
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Nov 16 18:13:18 UTC 2022 - 3.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/tf_data_fuse_map_and_batch.mlir
%3 = "tf.TensorSliceDataset"(%2) {device = "", output_shapes = [#tf_type.shape<>], metadata = ""} : (tensor<3xi32>) -> tensor<*x!tf_type.variant> // CHECK: "tf.MapAndBatchDataset"(%[[TSLICE]], %[[BSIZE:.*]], %[[NPC]] // CHECK-SAME: f = @"__inference_Dataset_map_<lambda>_80", %4 = "tf.MapDataset"(%3) {device = "", f = @"__inference_Dataset_map_<lambda>_80", output_shapes = [#tf_type.shape<>], output_types = [i32],
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 30 06:52:55 UTC 2023 - 1.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/utils/lstm_utils_test.cc
SmallVector<int64_t, 2> output_shape{1, mlir::ShapedType::kDynamic}; EXPECT_EQ(mlir::cast<RankedTensorType>(output_types[0]).getShape().size(), output_shape.size()); for (int i = 0; i < output_shape.size(); i++) { EXPECT_EQ(mlir::cast<RankedTensorType>(output_types[0]).getDimSize(i), output_shape[i]); } }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 10K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/mlir2graphdef/functional-while-ops.mlir
// CHECK: name: "StatefulWhile" // CHECK-NOT: name: // CHECK: op: "While" // CHECK-NOT: is_stateless // CHECK-NOT: shape_invariant // CHECK: attr { // CHECK: key: "output_shapes" // CHECK: value { // CHECK: list { // CHECK: shape { // CHECK: dim { // CHECK: size: 5 // CHECK: } // CHECK: } // CHECK: } // CHECK: }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Mar 25 12:28:56 UTC 2022 - 3.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/tf_data_fuse_pmap_and_batch.mlir
%4 = "tf.TensorSliceDataset"(%2) {device = "", output_shapes = [#tf_type.shape<>], metadata = ""} : (tensor<3xi32>) -> tensor<*x!tf_type.variant> // CHECK: "tf.MapAndBatchDataset"(%[[TSLICE]], // CHECK-SAME: f = @"__inference_Dataset_map_<lambda>_80", %5 = "tf.ParallelMapDataset"(%4, %3) {device = "", f = @"__inference_Dataset_map_<lambda>_80", output_shapes = [#tf_type.shape<>], output_types = [i32],
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Mar 24 05:47:26 UTC 2022 - 1.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_ops_canonicalization_helper.h
template <typename Op> struct DropAttributes : public OpRewritePattern<Op> { using OpRewritePattern<Op>::OpRewritePattern; // Drop the "output_shapes" attribute. LogicalResult matchAndRewrite(Op op, PatternRewriter &rewriter) const override { bool found = !!op->removeAttr("output_shapes"); return success(found); } }; // Helper function to create TF op while copying all underscore attributes from
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Nov 12 21:57:12 UTC 2021 - 2.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfr/tests/end2end.mlir
%drop_remainder: i1 {tfr.name="drop_remainder"}, %f: !tfr.attr {tfr.name="func"}, %output_types: !tfr.attr {tfr.name="output_types"}, %output_shapes: !tfr.attr {tfr.name="output_shapes"}, %preserve_cardinality: i1 {tfr.name="preserve_cardinality", tfr.default=false}) -> !tfr.tensor { %batch = "tfr.constant_tensor"(%batch_size) : (i64) -> tensor<i64>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 30 06:52:55 UTC 2023 - 13.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/decompose_reduce_dataset.mlir
) { // CHECK: tf.ReduceDataset %1 = "tf.ReduceDataset"(%arg0, %arg1) { Targuments = [], Tstate = [i64], device = "", f = @__reduce_func_0, f._tf_data_function = true, output_shapes = [#tf_type.shape<>], output_types = [i64], use_inter_op_parallelism = true } : (tensor<!tf_type.variant>, tensor<i64>) -> (tensor<i64>) func.return }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Aug 18 17:16:34 UTC 2022 - 9.8K bytes - Viewed (0)