| /external/tensorflow/tensorflow/cc/ops/ |
| D | while_loop.cc | 190 std::vector<Output> merge_outputs(num_loop_vars); in BuildWhileLoop() local
193 CreateMerge(scope, i, enter_outputs[i], &merge_outputs[i])); in BuildWhileLoop()
197 TF_RETURN_IF_ERROR(CreateCond(scope, cond, merge_outputs, &cond_out)); in BuildWhileLoop()
203 auto switch_i = Switch(scope, merge_outputs[i], cond_out); in BuildWhileLoop()
223 merge_outputs[i].node(), in BuildWhileLoop()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | bidirectional_sequence_rnn.cc | 98 params->merge_outputs ? 1 : 2); in Prepare()
354 params->merge_outputs ? fw_num_units + bw_num_units : fw_num_units; in Prepare()
357 if (!params->merge_outputs) { in Prepare()
415 params->merge_outputs ? fw_num_units + bw_num_units : fw_num_units; in EvalFloat()
417 params->merge_outputs ? fw_num_units + bw_num_units : bw_num_units; in EvalFloat()
447 (params->merge_outputs in EvalFloat()
487 params->merge_outputs in EvalFloat()
586 params->merge_outputs ? fw_num_units + bw_num_units : fw_num_units; in EvalHybrid()
588 params->merge_outputs ? fw_num_units + bw_num_units : bw_num_units; in EvalHybrid()
626 (params->merge_outputs in EvalHybrid()
[all …]
|
| D | bidirectional_sequence_lstm.cc | 428 params->merge_outputs ? 1 : 2); in Prepare()
554 params->merge_outputs ? n_bw_output + n_fw_output : n_fw_output; in Prepare()
611 if (!params->merge_outputs) { in Prepare()
1108 TfLiteTensor* bw_output = params->merge_outputs in Eval()
1149 params->merge_outputs ? fw_recurrent_to_output_weights->dims->data[1] : 0; in Eval()
1150 const auto actual_bw_output = params->merge_outputs ? fw_output : bw_output; in Eval()
|
| D | bidirectional_sequence_rnn_test.cc | 668 bool merge_outputs, bool quantize_weights = false, in BidirectionalRNNOpModel() argument
718 if (!merge_outputs) { in BidirectionalRNNOpModel()
726 merge_outputs, asymmetric_quantize_weights) in BidirectionalRNNOpModel()
|
| D | bidirectional_sequence_lstm_test.cc | 35 bool use_projection_bias, bool merge_outputs, in BidirectionalLSTMOpModel() argument
176 if (!merge_outputs) { in BidirectionalLSTMOpModel()
207 merge_outputs, time_major, asymmetric_quantize_inputs) in BidirectionalLSTMOpModel()
|
| /external/tensorflow/tensorflow/lite/c/ |
| D | builtin_op_data.h | 161 bool merge_outputs; member
298 bool merge_outputs; member
|
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | group_bidirectional_sequence_ops.cc | 251 (*bi_op)->merge_outputs = false; in ConstructBidirectionalSequenceOp()
299 (*bi_op)->merge_outputs = false; in ConstructBidirectionalSequenceOp()
351 bidirectional_sequence_ops[i]->merge_outputs = false; in RewireBidirectionalSequenceSequenceOpsConnections()
355 bidirectional_sequence_ops[i]->merge_outputs = true; in RewireBidirectionalSequenceSequenceOpsConnections()
|
| D | propagate_fixed_sizes.cc | 1222 if (op->merge_outputs) { in ProcessBidirectionalSequenceLstmOperator()
1274 if (op->merge_outputs) { in ProcessBidirectionalSequenceRnnOperator()
|
| /external/tensorflow/tensorflow/lite/toco/tflite/ |
| D | operator.cc | 934 /*merge_outputs=*/op.merge_outputs, in WriteOptions()
943 op->merge_outputs = options.merge_outputs(); in ReadOptions()
976 /*merge_outputs=*/op.merge_outputs); in WriteOptions()
984 op->merge_outputs = options.merge_outputs(); in ReadOptions()
|
| /external/tensorflow/tensorflow/lite/core/api/ |
| D | flatbuffer_conversions.cc | 513 params->merge_outputs = bidi_sequence_rnn_params->merge_outputs(); in ParseOpDataTfLite()
619 params->merge_outputs = bidi_lstm_params->merge_outputs(); in ParseOpDataTfLite()
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | model.h | 700 bool merge_outputs;
706 bool merge_outputs;
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/tests/flatbuffer2mlir/ |
| D | test_schema.fbs | 544 merge_outputs: bool;
643 merge_outputs: bool;
|
| /external/tensorflow/tensorflow/lite/schema/ |
| D | schema_v3a.fbs | 548 merge_outputs: bool;
649 merge_outputs: bool;
|
| D | schema.fbs | 576 merge_outputs: bool;
677 merge_outputs: bool;
|
| D | schema_generated.h | 4720 bool merge_outputs;
4725 merge_outputs(false),
4744 bool merge_outputs() const {
4772 void add_merge_outputs(bool merge_outputs) {
4773 …int8_t>(BidirectionalSequenceRNNOptions::VT_MERGE_OUTPUTS, static_cast<uint8_t>(merge_outputs), 0);
4794 bool merge_outputs = false,
4798 builder_.add_merge_outputs(merge_outputs);
5489 bool merge_outputs;
5496 merge_outputs(false),
5521 bool merge_outputs() const {
[all …]
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/tests/ |
| D | ops.mlir | 748 … {cell_clip = 1.000000e+00 : f32, fused_activation_function = "NONE", merge_outputs = true, time_m…
749 … {cell_clip = 1.000000e+00 : f32, fused_activation_function = "NONE", merge_outputs = true, time_m…
|
| /external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| D | nnapi_delegate.cc | 3522 mapping_args.builder->AddScalarBoolOperand(builtin->merge_outputs); in Map()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/ir/ |
| D | tfl_ops.td | 4195 BoolAttr:$merge_outputs,
|