- Sort Score
- Result 10 results
- Languages All
Results 21 - 30 of 52 for _output_shapes (0.31 sec)
-
tensorflow/compiler/mlir/tf2xla/api/v1/compile_mlir_util_test.cc
// Create a bad output shape attr. AttrValue shape_attr; TensorShapeProto* shape_proto = shape_attr.mutable_list()->add_shape(); shape_proto->add_dim()->set_size(1); builder.Attr("_output_shapes", shape_attr); TF_RETURN_IF_ERROR(builder.Finalize(&node)); return CreateSingleOpGraph(node, {}, {DataType::DT_INT32}); } absl::Status BuildHloFromGraph(Graph& graph, bool use_output_shapes) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Mar 25 19:54:38 UTC 2024 - 9.7K 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/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/jit/encapsulate_util_test.cc
auto node_index = g.BuildNodeNameIndex(); Node *add_node = node_index["add"]; std::vector<PartialTensorShape> output_shapes; TF_CHECK_OK(GetNodeAttr(add_node->attrs(), kXlaInferredShapesAttrName, &output_shapes)); EXPECT_EQ(output_shapes.size(), 1); TensorShapeProto shape_proto; output_shapes[0].AsProto(&shape_proto); EXPECT_EQ(shape_proto.dim_size(), 1); EXPECT_EQ(shape_proto.dim(0).size(), 2); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Feb 05 22:06:22 UTC 2019 - 2.2K 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)