Searched refs:merge_outputs (Results 1 – 13 of 13) sorted by relevance
| /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 | 94 params->merge_outputs ? 1 : 2); in Prepare()
266 params->merge_outputs ? fw_num_units + bw_num_units : fw_num_units; in Prepare()
269 if (!params->merge_outputs) { in Prepare()
321 params->merge_outputs ? fw_num_units + bw_num_units : fw_num_units; in EvalFloat()
323 params->merge_outputs ? fw_num_units + bw_num_units : bw_num_units; in EvalFloat()
353 (params->merge_outputs ? fw_output->data.f + fw_num_units in EvalFloat()
390 params->merge_outputs in EvalFloat()
482 params->merge_outputs ? fw_num_units + bw_num_units : fw_num_units; in EvalHybrid()
484 params->merge_outputs ? fw_num_units + bw_num_units : bw_num_units; in EvalHybrid()
519 (params->merge_outputs ? fw_output->data.f + fw_num_units in EvalHybrid()
[all …]
|
| D | bidirectional_sequence_lstm.cc | 394 params->merge_outputs ? 1 : 2); in Prepare()
507 params->merge_outputs ? n_bw_output + n_fw_output : n_fw_output; in Prepare()
561 if (!params->merge_outputs) { in Prepare()
847 TfLiteTensor* bw_output = params->merge_outputs in Eval()
885 params->merge_outputs ? fw_recurrent_to_output_weights->dims->data[1] : 0; in Eval()
886 const auto actual_bw_output = params->merge_outputs ? fw_output : bw_output; in Eval()
|
| D | bidirectional_sequence_rnn_test.cc | 658 bool time_major, bool merge_outputs) in BidirectionalRNNOpModel() argument
685 if (!merge_outputs) { in BidirectionalRNNOpModel()
693 builder_, time_major, ActivationFunctionType_RELU, merge_outputs) in BidirectionalRNNOpModel()
|
| D | bidirectional_sequence_lstm_test.cc | 40 bool use_projection_bias, bool merge_outputs, in BidirectionalLSTMOpModel() argument
184 if (!merge_outputs) { in BidirectionalLSTMOpModel()
206 proj_clip, merge_outputs, time_major) in BidirectionalLSTMOpModel()
|
| /external/tensorflow/tensorflow/lite/c/ |
| D | builtin_op_data.h | 120 bool merge_outputs; member
224 bool merge_outputs; member
|
| /external/tensorflow/tensorflow/lite/core/api/ |
| D | flatbuffer_conversions.cc | 256 params->merge_outputs = bidi_sequence_rnn_params->merge_outputs(); in ParseOpData()
426 params->merge_outputs = bidi_lstm_params->merge_outputs(); in ParseOpData()
|
| /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()
349 bidirectional_sequence_ops[i]->merge_outputs = false; in RewireBidirectionalSequenceSequenceOpsConnections()
353 bidirectional_sequence_ops[i]->merge_outputs = true; in RewireBidirectionalSequenceSequenceOpsConnections()
|
| D | propagate_fixed_sizes.cc | 1209 if (op->merge_outputs) { in ProcessBidirectionalSequenceLstmOperator()
1261 if (op->merge_outputs) { in ProcessBidirectionalSequenceRnnOperator()
|
| /external/tensorflow/tensorflow/lite/toco/tflite/ |
| D | operator.cc | 1118 /*merge_outputs=*/op.merge_outputs, in WriteOptions()
1127 op->merge_outputs = options.merge_outputs(); in ReadOptions()
1164 /*merge_outputs=*/op.merge_outputs); in WriteOptions()
1172 op->merge_outputs = options.merge_outputs(); in ReadOptions()
|
| /external/tensorflow/tensorflow/lite/schema/ |
| D | schema.fbs | 399 merge_outputs: bool;
481 merge_outputs: bool;
|
| D | schema_generated.h | 3456 bool merge_outputs;
3460 merge_outputs(false) {
3477 bool merge_outputs() const {
3501 void add_merge_outputs(bool merge_outputs) {
3502 …int8_t>(BidirectionalSequenceRNNOptions::VT_MERGE_OUTPUTS, static_cast<uint8_t>(merge_outputs), 0);
3520 bool merge_outputs = false) {
3522 builder_.add_merge_outputs(merge_outputs);
4153 bool merge_outputs;
4159 merge_outputs(false),
4182 bool merge_outputs() const {
[all …]
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | model.h | 664 bool merge_outputs;
670 bool merge_outputs;
|