- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 11 for _arg0 (0.12 sec)
-
tensorflow/compiler/mlir/tensorflow/tests/shape_inference.mlir
} func.func @partitioned_called_func(%arg0: tensor<?xi32>) -> (tensor<?xi32>) { func.return %arg0 : tensor<?xi32> } func.func @stateful_partitioned_call_func(%arg0: tensor<?xi32>) -> (tensor<?xi32>) { func.return %arg0 : tensor<?xi32> } func.func @tpu_partitioned_call_func(%arg0: tensor<?xi32>) -> (tensor<?xi32>) { func.return %arg0 : tensor<?xi32> }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jan 23 17:24:10 UTC 2024 - 167.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/canonicalize.mlir
%0 = "tf.Bitcast"(%arg0) : (tensor<8x16x32x64xf32>) -> tensor<8x16x32x64xf32> // CHECK: return %arg0 func.return %0: tensor<8x16x32x64xf32> } // CHECK-LABEL: testDifferentBitcastType func.func @testDifferentBitcastType(%arg0: tensor<8x16x32x64xf32>) -> tensor<8x16x32x64xi32> {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 09 22:07:10 UTC 2024 - 132.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/tests/ops.mlir
^bb0(%arg0: tensor<? x f32>): // CHECK: "tfl.log"(%arg0) %0 = "tfl.log"(%arg0): (tensor<? x f32>) -> tensor<? x f32> func.return %0 : tensor<? x f32> } // CHECK-LABEL: testLogStaticShapeInputAndDynamicShapeOutput func.func @testLogStaticShapeInputAndDynamicShapeOutput(tensor<8 x f32>) -> tensor<? x f32> { ^bb0(%arg0: tensor<8 x f32>): // CHECK: "tfl.log"(%arg0)
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 06 19:09:08 UTC 2024 - 189.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/tests/legalize-tf.mlir
// CHECK: %3 = "tf.MaxPool"(%arg0) // CHECK: %4 = "tf.MaxPool"(%arg0) } func.func @abs(%arg0: tensor<8x16xf32>) -> tensor<8x16xf32> { %0 = "tf.Abs"(%arg0) : (tensor<8x16xf32>) -> tensor<8x16xf32> func.return %0 : tensor<8x16xf32> // CHECK-LABEL:abs // CHECK: "tfl.abs"(%arg0) : (tensor<8x16xf32>) -> tensor<8x16xf32> } func.func @any(%arg0: tensor<2x2xi1>, %arg1: tensor<i32>) -> tensor<i1> {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 05 01:54:33 UTC 2024 - 153.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/tpu_rewrite.mlir
func.func @replicated_model_parallel_execute_tpu_cluster_func(%arg0: tensor<?xi32>) -> tensor<?xi32> { // CHECK: %[[A_OUTPUT:[0-9]*]] = "tf.A" %0 = "tf.A"(%arg0) : (tensor<?xi32>) -> tensor<?xi32> // CHECK: %[[REPLICATE:[0-9]*]]:4 = tf_device.replicate %1:4 = tf_device.replicate([%0, %arg0, %arg0, %arg0] as %ri_0: tensor<?xi32>) {n = 4 : i32} {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 22:03:30 UTC 2024 - 172.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/tests/uniform-quantized-stablehlo-to-tfl.mlir
func.func @reshape_float(%arg0: tensor<2x3x4xf32>) -> tensor<6x4xf32> { %0 = stablehlo.reshape %arg0 : (tensor<2x3x4xf32>) -> tensor<6x4xf32> return %0 : tensor<6x4xf32> } // CHECK-LABEL: reshape_float // CHECK-NOT: tfl.reshape // CHECK: stablehlo.reshape // ----- // Tests that a quantized `stablehlo.select` is converted to `tfl.select_v2`. func.func @select( %arg0: tensor<1x3xi1>,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 14 17:10:32 UTC 2024 - 106.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/tests/prepare-composite-functions-tf.mlir
module { // CHECK-LABEL: func private @some_func // CHECK-LABEL: func @func_with_call func.func private @some_func(%arg0: tensor<100xf32>) -> tensor<100xf32> attributes {tf.api_implements = "lstm_b4e9f0e7-ac55-42bc-8ef2-8496419a608c"} func.func @func_with_call(%arg0: tensor<100xf32>) -> tensor<100xf32> { %0 = func.call @some_func(%arg0) : (tensor<100xf32>) -> tensor<100xf32> func.return %0 : tensor<100xf32> } } // -----
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 09:41:17 UTC 2024 - 122.1K bytes - Viewed (0) -
tensorflow/compiler/jit/extract_outside_compilation_pass.cc
} // Builds XlaRecvAtHost node, and replaces all _Arg nodes with it. absl::StatusOr<Node*> ReplaceArgNodesWithRecvAtHostNode( Graph* g, const string& oc_cluster_name, std::vector<DataType>* recv_at_host_dtypes, Node* key_placeholder) { // TODO(b/77601805): use out nodes for source node, instead of traversing all // nodes. std::vector<Node*> arg_nodes = GatherNodesWithType(*g, "_Arg");
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 12 06:33:33 UTC 2024 - 104.7K bytes - Viewed (0) -
tensorflow/compiler/jit/encapsulate_subgraphs_pass_test.cc
node_builder.Input(inputs); return opts.FinalizeBuilder(&node_builder); } Node* ArgOp(int index, DataType type, const GraphDefBuilder::Options& opts) { return ops::SourceOp("_Arg", opts.WithAttr("T", type).WithAttr("index", index)); } Node* RetOp(int index, ops::NodeOut a, const GraphDefBuilder::Options& opts) { if (opts.HaveError()) return nullptr;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 08:47:20 UTC 2024 - 113.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/translate/import_model.cc
return errors::InvalidArgument( "Invalid \"_handle_dtypes\" attribute value for _Arg node: ", shape_attr->DebugString()); } if (shape_attr->list().shape().empty()) { return errors::InvalidArgument( "Invalid \"_handle_shapes\" attribute value for _Arg node: ", shape_attr->DebugString()); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 01 11:17:36 UTC 2024 - 183.2K bytes - Viewed (0)