- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 402 for _arg0 (0.05 sec)
-
tensorflow/compiler/jit/xla_compile_util_test.cc
const Node* identity_input_node = nullptr; TF_EXPECT_OK(identity_node->input_node(0, &identity_input_node)); EXPECT_EQ(identity_input_node->name(), "_arg0"); const Node* arg_node = node_name_index.at("_arg0"); EXPECT_EQ(arg_node->op_def().name(), "_Arg"); EXPECT_EQ(arg_node->attrs().FindByString("T")->type(), DT_FLOAT); const Node* retval_node = node_name_index.at("_retval0");
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 16 21:48:05 UTC 2023 - 6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/tf2xla_rewriter_test.cc
func.func @main(%arg0: tensor<3x3x10xbf16>, %arg1: tensor<3xi32>) -> tensor<1x?x4xbf16> attributes {allow_soft_placement = false, tf.entry_function = {control_outputs = "", inputs = "_arg0,_arg1,_arg2", outputs = "_retval0"}} { %cst = "tf.Const"() {value = dense<[1, -1, 4]> : tensor<3xi32>} : () -> tensor<3xi32>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 09:16:07 UTC 2024 - 11.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/decompose_resource_ops.mlir
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 22 19:47:48 UTC 2024 - 51.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/tests/legalize-tf.mlir
%0:2 = "tf.IdentityN"(%arg0, %arg1) : (tensor<1xi32>, tensor<1xf32>) -> (tensor<1xi32>, tensor<1xf32>) func.return %0#0, %0#1: tensor<1xi32>, tensor<1xf32> } // ----- // CHECK-LABEL: func @stopgradient func.func @stopgradient(%arg0: tensor<1xi32>) -> tensor<1xi32> { // CHECK-NEXT: return %arg0 : tensor<1xi32> %0 = "tf.StopGradient"(%arg0) : (tensor<1xi32>) -> tensor<1xi32>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon May 06 18:46:23 UTC 2024 - 335.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/stack_ops_decomposition.mlir
%while = "tf.WhileRegion"(%max_size) ({ // CHECK: ^bb0(%[[BARG0:.*]]: tensor<i32> ^bb0(%barg0: tensor<i32>): // CHECK: "tf._SomeOp"(%[[BARG0]]) %pred = "tf._SomeOp"(%barg0) : (tensor<i32>) -> tensor<i1> "tf.Yield"(%pred) : (tensor<i1>) -> () }, { // CHECK: ^bb0(%[[BARG0:.*]]: tensor<i32> ^bb0(%barg0: tensor<i32>):
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 30 06:52:55 UTC 2023 - 25.8K bytes - Viewed (0) -
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/tensor_array_ops_decomposition.mlir
// CHECK: return %[[BARG0]], %[[SUB]], %[[BARG2]] func.return %arg0, %sub : tensor<!tf_type.resource>, tensor<i32> } // CHECK: func @while_cond(%[[CARG0:.*]]: tensor<!tf_type.resource<tensor<5x3xf32>>>, %[[CARG1:.*]]: tensor<i32>, %[[CARG2:.*]]: tensor<!tf_type.resource<tensor<5x3xf32>>>) func.func @while_cond(%arg0: tensor<!tf_type.resource>, %arg1: tensor<i32>) -> tensor<i32> {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 30 06:52:55 UTC 2023 - 49K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/resource_op_lifting.mlir
func.return } // CHECK: func @while_body(%[[BARG0:.*]]: tensor<f32>) func.func @while_body(%arg0: tensor<*x!tf_type.resource<tensor<f32>>>) -> (tensor<*x!tf_type.resource<tensor<f32>>>) { // CHECK-NEXT: return %[[BARG0]] func.return %arg0 : tensor<*x!tf_type.resource<tensor<f32>>> } // CHECK: func @while_cond(%[[CARG0:.*]]: tensor<f32>) func.func @while_cond(%arg0: tensor<*x!tf_type.resource<tensor<f32>>>) -> tensor<f32> {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 30 06:52:55 UTC 2023 - 74K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/region-control-flow-to-functional.mlir
{ ^bb0(%carg0: tensor<*xf32>, %carg1: tensor<i32>): %ne = "tf.NotEqual"(%carg1, %zero) : (tensor<i32>, tensor<i32>) -> tensor<i1> "tf.Yield"(%ne) : (tensor<i1>) -> () }, { // loop body ^bb0(%barg0: tensor<*xf32>, %barg1: tensor<i32>): %add = "tf.Add"(%barg0, %barg0) : (tensor<*xf32>, tensor<*xf32>) -> tensor<*xf32>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Feb 02 11:15:34 UTC 2024 - 44.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/remove_unused_while_results.mlir
// CHECK-LABEL: result_used_in_cond // CHECK: tf.WhileRegion // CHECK-SAME: (%[[ARG0:[a-zA-Z0-9_]+]], %[[ARG1:[a-zA-Z0-9_]+]]) // CHECK: ^bb0 // CHECK: (%[[CARG0:[a-zA-Z0-9_]+]]: tensor<*xf32>, %[[CARG1:[a-zA-Z0-9_]+]]: tensor<*xf32>) // CHECK: tf.OpA // CHECK: ^bb0 // CHECK: (%[[BARG0:[a-zA-Z0-9_]+]]: tensor<*xf32>, %[[BARG1:[a-zA-Z0-9_]+]]: tensor<*xf32>) // CHECK: tf.OpB
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 21 20:25:31 UTC 2022 - 8.5K bytes - Viewed (0)