Home
last modified time | relevance | path

Searched refs:num_outs (Results 1 – 12 of 12) sorted by relevance

/external/tensorflow/tensorflow/core/ops/
Dcontrol_flow_ops.cc48 int num_outs; in SwitchNShape() local
49 TF_RETURN_IF_ERROR(c->GetAttr("num_outs", &num_outs)); in SwitchNShape()
50 for (int i = 0; i < num_outs; i++) { in SwitchNShape()
57 for (int i = 0; i < num_outs; i++) { in SwitchNShape()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_executor.cc464 IntegerAttr num_outs = switchn->getAttrOfType<IntegerAttr>("num_outs"); in Verify() local
465 if (!num_outs) in Verify()
469 if (switchn.getNumResults() != num_outs.getInt() + 1) in Verify()
471 << "expect `num_outs` (" << num_outs.getInt() << ") results but got " in Verify()
540 IntegerAttr num_outs; in ParseSwitchNOp() local
543 parser.parseAttribute(num_outs, i64_type, "num_outs", in ParseSwitchNOp()
553 if (num_outs.getInt() <= 0) in ParseSwitchNOp()
568 result.types.append(num_outs.getInt(), types[0]); in ParseSwitchNOp()
Dtf_executor_ops.td295 and an integer attribute `num_outs` indicating the number of outputs. The
307 .Output("outputs: num_outs * T")
308 .Attr("num_outs: int >= 1")
323 I64Attr:$num_outs
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/mlir2graphdef/
Dswitchn.mlir10 …l_1 = "tf_executor._SwitchN"(%outputs, %outputs) {T = i32, device = "", num_outs = 3 : i64} : (ten…
39 …outputs_14, %outputs) {T = f32, _class = ["Case/input_0"], device = "", num_outs = 2 : i64} : (ten…
/external/tensorflow/tensorflow/python/framework/
Dpython_op_gen_internal.cc287 const int num_outs = op_def.output_arg_size(); in GetReturns() local
289 if (num_outs == 0) { in GetReturns()
292 if (num_outs == 1) { in GetReturns()
317 std::vector<string> out_names(num_outs); in GetReturns()
318 for (int i = 0; i < num_outs; ++i) { in GetReturns()
327 for (int i = 0; i < num_outs; ++i) { in GetReturns()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/graphdef2mlir/
Dswitch_n.pbtxt66 key: "num_outs"
92 key: "num_outs"
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/
Dtf_executor_ops_invalid.mlir423 …%1:3 = "tf_executor._SwitchN"(%arg1, %arg0) {num_outs = 5} : (tensor<*xf32>, tensor<i32>) -> (tens…
424 // expected-error@-1 {{'tf_executor._SwitchN' op expect `num_outs` (5) results but got 2}}
436 …%1:3 = "tf_executor._SwitchN"(%arg0, %arg1) {num_outs = 2} : (i32, tensor<i32>) -> (tensor<*xi32>,…
448 …%1:3 = "tf_executor._SwitchN"(%arg0, %arg1) {num_outs = 2} : (tensor<*xi32>, tensor<i32>) -> (i32,…
461 …%1:3 = "tf_executor._SwitchN"(%arg0, %arg1) {num_outs = 2} : (tensor<4xf32>, tensor<i32>) -> (tens…
474 …%1:3 = "tf_executor._SwitchN"(%arg0, %arg1) {num_outs = 2} : (tensor<*xf32>, tensor<i32>) -> (tens…
Dshape_inference.mlir360 …%switchn:3 = "tf_executor._SwitchN"(%island#0, %arg3) {num_outs = 2} : (tensor<?x?x?xf32>, tensor<…
/external/tensorflow/tensorflow/compiler/jit/
Ddeadness_analysis_test.cc1224 int64 num_outs, OutputList* outputs) { in CreateSwitchN() argument
1235 .Attr("num_outs", num_outs); in CreateSwitchN()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dlayout_optimizer.cc2005 const int num_outs = in GetOutputPos() local
2007 for (int i = 0; i < num_outs; i++) { in GetOutputPos()
Dgeneric_layout_optimizer_transposer.cc2151 const int num_outs = num_outs_attr != nullptr ? num_outs_attr->i() : 2; in GetDataFanoutPorts() local
2152 std::vector<int> values(num_outs); in GetDataFanoutPorts()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/g3doc/
Dtf_dialects.md230 takes two inputs,`data`and`index`and an integer attribute`num_outs`indicating