- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 86 for output_shapes (0.34 sec)
-
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/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/lite/tests/lift_tflite_flex_ops.mlir
} : (tensor<!tf_type.variant>) -> tensor<!tf_type.variant> func.return %0 : tensor<!tf_type.variant> // CHECK: "tf.MapDataset"( // CHECK-SAME: <{f = @{{.*}}, metadata = "", output_shapes = [#tf_type.shape<>], output_types = [!tf_type.string], preserve_cardinality = true, use_inter_op_parallelism = true}> {Targuments = []} } // CHECK-LABEL: TfTakeWhileDataset
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 30 06:52:55 UTC 2023 - 6.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tf_data_optimization.td
def FuseMapAndBatch : Pat< (TF_BatchDatasetV2Op (TF_MapDatasetOp $input_dataset, $other_arguments, $f, $output_types, $output_shapes, $use_inter_op_parallelism, $preserve_cardinality, $force_synchronous, $map_dataset_metadata), $batch_size, $drop_remainder, $parallel_copy, $batch_output_types,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 23:24:08 UTC 2024 - 1.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/resource-alias-analysis-test.mlir
// expected-remark@below {{Result #0, ID 0 : 0, 1}} %iter_handle1 = "tf.IteratorV2"() {container = "c", shared_name = "v0", output_shapes = [#tf_type.shape<>], output_types = [!tf_res]} : () -> !tf_res // expected-remark@below {{Result #0, ID 1 : 0, 1}} %iter_handle2 = "tf.IteratorV2"() {container = "c", shared_name = "v1", output_shapes = [#tf_type.shape<>], output_types = [!tf_res]} : () -> !tf_res func.return } // -----
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Jul 21 17:19:47 UTC 2023 - 22.2K bytes - Viewed (0) -
tensorflow/compiler/jit/encapsulate_util.cc
std::vector<PartialTensorShape> output_shapes; std::transform(iter.second.begin(), iter.second.end(), std::back_inserter(output_shapes), [](const InferredShape& inferred_shape) { return inferred_shape.shape; }); Node* n = node_name_index[iter.first]; n->AddAttr(kXlaInferredShapesAttrName, output_shapes); } return absl::OkStatus();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 12 06:33:33 UTC 2024 - 15.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/tests/fuse-tftext.mlir
%17 = "tf.If"(%2, %2, %13, %13) {_lower_using_switch_merge = true, _read_only_resource_inputs = [], device = "", else_branch = @WhitespaceTokenize_RaggedConcat_assert_equal_1_Assert_AssertGuard_false_3210, is_stateless = false, output_shapes = [#tf_type.shape<>], then_branch = @WhitespaceTokenize_RaggedConcat_assert_equal_1_Assert_AssertGuard_true_3200} : (tensor<i1>, tensor<i1>, tensor<i64>, tensor<i64>) -> tensor<i1>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 09:41:17 UTC 2024 - 460.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/translate/export_tf_dialect_op.cc
end++; if (begin != end) { mlir::TF::ResultShapeRange output_shapes = { mlir::TF::ResultShapeIterator(begin), mlir::TF::ResultShapeIterator(end)}; SetShapeAttribute("_output_shapes", output_shapes, attributes); } } return absl::OkStatus(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 01 11:17:36 UTC 2024 - 11.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/decompose_optionals.mlir
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 21:18:05 UTC 2024 - 4.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/functional-control-flow-to-regions.mlir
device = "/job:tpu_host_worker/replica:0/task:0/device:CPU:0", finalize_func = @finalize, init_func = @init, next_func = @next, operandSegmentSizes = array<i32: 1, 2, 1>, output_shapes = [#tf_type.shape<>], output_types = [!tf_type.string], metadata = ""} : ( tensor<4xf32>, tensor<3xf32>, tensor<!tf_type.resource>,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Nov 06 21:59:28 UTC 2023 - 11.9K bytes - Viewed (0)