| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | group_bidirectional_sequence_ops.cc | 87 Operator* op_it = nullptr; in FindUnidirectionalSequenceOp() local
88 op_it = GetOpWithOutput(model, output_op.inputs[0]); in FindUnidirectionalSequenceOp()
89 if (op_it == nullptr) { in FindUnidirectionalSequenceOp()
93 while (op_it->type == operator_type) { in FindUnidirectionalSequenceOp()
94 sequence_ops->push(op_it); in FindUnidirectionalSequenceOp()
96 op_it = GetOpWithOutput(model, op_it->inputs[0]); in FindUnidirectionalSequenceOp()
97 if (op_it == nullptr) { in FindUnidirectionalSequenceOp()
102 *input_op = op_it; in FindUnidirectionalSequenceOp()
323 std::vector<std::unique_ptr<Operator>>::iterator* op_it, Model* model) { in RewireBidirectionalSequenceSequenceOpsConnections() argument
359 model->operators.emplace(*op_it, bidirectional_sequence_ops[i]); in RewireBidirectionalSequenceSequenceOpsConnections()
[all …]
|
| D | unpartition_embedding_lookup.cc | 51 auto op_it = model->operators.begin() + op_index; in Run() local
52 if (op_it->get()->type != OperatorType::kDynamicStitch) { in Run()
55 auto* stitch_op = static_cast<DynamicStitchOperator*>(op_it->get()); in Run()
183 op_it = model->operators.emplace(op_it, gather_params_concat_op) + 1; in Run()
196 op_it = model->operators.emplace(op_it, gather_params_permute_op) + 1; in Run()
222 model->operators.emplace(op_it, merged_gather_op); in Run()
|
| D | resolve_transpose_attributes.cc | 31 const auto op_it = model->operators.begin() + op_index; in Run() local
32 if (op_it->get()->type != OperatorType::kTranspose) in Run()
35 auto* op = static_cast<TransposeOperator*>(op_it->get()); in Run()
|
| D | resolve_batch_to_space_nd_attributes.cc | 31 const auto op_it = model->operators.begin() + op_index; in Run() local
32 if (op_it->get()->type != OperatorType::kBatchToSpaceND) in Run()
35 auto* op = static_cast<BatchToSpaceNDOperator*>(op_it->get()); in Run()
|
| D | identify_relu1.cc | 34 const auto op_it = model->operators.begin() + op_index; in Run() local
35 const auto* op_0 = op_it->get(); in Run()
76 model->operators.emplace(op_it, relu1_op); in Run()
|
| D | resolve_space_to_batch_nd_attributes.cc | 31 const auto op_it = model->operators.begin() + op_index; in Run() local
32 if (op_it->get()->type != OperatorType::kSpaceToBatchND) in Run()
35 auto* op = static_cast<SpaceToBatchNDOperator*>(op_it->get()); in Run()
|
| D | remove_trivial_fake_quant.cc | 71 const auto op_it = model->operators.begin() + op_index; in Run() local
72 auto* op = op_it->get(); in Run()
|
| D | identify_lstm_split_inputs.cc | 33 auto op_it = model->operators.begin() + op_index; in Run() local
34 auto curr_op = op_it->get(); in Run()
163 model->operators.emplace(op_it, std::move(lstm_cell_op)); in Run()
|
| D | identify_nearest_upsample.cc | 85 auto op_it = model->operators.begin() + op_index; in Run() local
86 auto* op = op_it->get(); in Run()
271 op_it = model->operators.emplace(op_it, to_be_inserted_ops.at(i)); in Run()
|
| D | identify_lstm_merge_inputs.cc | 33 auto op_it = model->operators.begin() + op_index; in Run() local
34 auto src_op = op_it->get(); in Run()
170 model->operators.emplace(op_it, std::move(lstm_cell_op)); in Run()
|
| D | identify_lstm.cc | 145 auto op_it = model->operators.begin() + op_index; in Run() local
146 Operator* final_output_mul = op_it->get(); in Run()
269 model->operators.emplace(op_it, lstm_cell_op); in Run()
|
| D | dequantize.cc | 192 const auto op_it = model->operators.begin() + op_index; in Run() local
193 auto* op = op_it->get(); in Run()
|
| D | quantize.cc | 720 const auto op_it = FindOp(*model, &op); in Run() local
721 model->operators.emplace(op_it + 1, dequantize_op); in Run()
|
| /external/tensorflow/tensorflow/c/eager/ |
| D | tape.h | 458 auto op_it = op_tape_.find(op_id); in DeleteTrace() local
459 CHECK(op_it != op_tape_.end()); in DeleteTrace()
460 for (const auto& output : op_it->second.output_tensor_info) { in DeleteTrace()
466 for (int64 id : op_it->second.input_tensor_id) { in DeleteTrace()
469 op_it->second.backward_function_deleter(op_it->second.backward_function); in DeleteTrace()
470 op_tape_.erase(op_it); in DeleteTrace()
541 auto op_it = op_tape->find(op_id); in PrepareBackprop() local
543 if (op_id == -1 || op_it == op_tape->end() || in PrepareBackprop()
547 CHECK(result.op_tape.emplace(op_id, op_it->second).second); in PrepareBackprop()
548 for (auto it : op_it->second.input_tensor_id) { in PrepareBackprop()
[all …]
|
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/tests/ |
| D | identify_l2_pool_test.cc | 66 for (auto& op_it : model.operators) { in RunIdentifyL2Pool() local
67 Operator* op = op_it.get(); in RunIdentifyL2Pool()
|
| D | identify_l2_normalization_test.cc | 97 for (auto& op_it : model.operators) { in RunIdentifyL2Normalization() local
98 Operator* op = op_it.get(); in RunIdentifyL2Normalization()
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | tooling_util.cc | 196 auto op_it = FindOp(*model, op); in DeleteOpAndArrays() local
197 CHECK(op_it != model->operators.end()); in DeleteOpAndArrays()
198 model->operators.erase(op_it); in DeleteOpAndArrays()
288 for (auto& op_it : model->operators) { in ReplaceArrayUsage() local
289 Operator* op = op_it.get(); in ReplaceArrayUsage()
|