Searched refs:num_outs (Results 1 – 12 of 12) sorted by relevance
/external/tensorflow/tensorflow/core/ops/ |
D | control_flow_ops.cc | 48 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/ |
D | tf_executor.cc | 464 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()
|
D | tf_executor_ops.td | 295 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/ |
D | switchn.mlir | 10 …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/ |
D | python_op_gen_internal.cc | 287 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/ |
D | switch_n.pbtxt | 66 key: "num_outs" 92 key: "num_outs"
|
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/ |
D | tf_executor_ops_invalid.mlir | 423 …%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…
|
D | shape_inference.mlir | 360 …%switchn:3 = "tf_executor._SwitchN"(%island#0, %arg3) {num_outs = 2} : (tensor<?x?x?xf32>, tensor<…
|
/external/tensorflow/tensorflow/compiler/jit/ |
D | deadness_analysis_test.cc | 1224 int64 num_outs, OutputList* outputs) { in CreateSwitchN() argument 1235 .Attr("num_outs", num_outs); in CreateSwitchN()
|
/external/tensorflow/tensorflow/core/grappler/optimizers/ |
D | layout_optimizer.cc | 2005 const int num_outs = in GetOutputPos() local 2007 for (int i = 0; i < num_outs; i++) { in GetOutputPos()
|
D | generic_layout_optimizer_transposer.cc | 2151 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/ |
D | tf_dialects.md | 230 takes two inputs,`data`and`index`and an integer attribute`num_outs`indicating
|