- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 46 for output_shapes (0.42 sec)
-
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/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/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/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/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/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) -
tensorflow/compiler/mlir/tensorflow/tests/graphdef2mlir/tf-data-pipeline.pbtxt
op: "TensorSliceDataset" input: "tensors/normalize_tensors/component_0" attr { key: "Toutput_types" value { list { type: DT_INT32 } } } attr { key: "output_shapes" value { list { shape { } } } } } node { name: "MapDataset" op: "MapDataset" input: "TensorSliceDataset" attr { key: "Targuments"
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jul 29 04:41:05 UTC 2021 - 4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/mlir2graphdef/case.mlir
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Mar 28 12:06:33 UTC 2022 - 2.5K bytes - Viewed (0)