| /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
357 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()
181 op_it = model->operators.emplace(op_it, gather_params_concat_op) + 1; in Run()
194 op_it = model->operators.emplace(op_it, gather_params_permute_op) + 1; in Run()
220 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 | 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 | identify_relu1.cc | 63 const auto op_it = model->operators.begin() + op_index; in Run() local
64 const auto* op_0 = op_it->get(); in Run()
104 model->operators.emplace(op_it, relu1_op); 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_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()
169 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 | 675 const auto op_it = FindOp(*model, &op); in Run() local
676 model->operators.emplace(op_it + 1, dequantize_op); in Run()
|
| /external/tensorflow/tensorflow/c/eager/ |
| D | tape.h | 258 auto op_it = op_tape_.find(op_id); in DeleteTrace() local
259 CHECK(op_it != op_tape_.end()); in DeleteTrace()
260 for (const auto& output : op_it->second.output_tensor_info) { in DeleteTrace()
266 for (int64 id : op_it->second.input_tensor_id) { in DeleteTrace()
269 op_it->second.backward_function_deleter(op_it->second.backward_function); in DeleteTrace()
270 op_tape_.erase(op_it); in DeleteTrace()
341 auto op_it = op_tape->find(op_id); in PrepareBackprop() local
343 if (op_id == -1 || op_it == op_tape->end() || in PrepareBackprop()
347 CHECK(result.op_tape.emplace(op_id, op_it->second).second); in PrepareBackprop()
348 for (auto it : op_it->second.input_tensor_id) { in PrepareBackprop()
[all …]
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | tooling_util.cc | 180 auto op_it = FindOp(*model, op); in DeleteOpAndArraysIfUnused() local
181 CHECK(op_it != model->operators.end()); in DeleteOpAndArraysIfUnused()
182 model->operators.erase(op_it); in DeleteOpAndArraysIfUnused()
271 for (auto& op_it : model->operators) { in ReplaceArrayUsage() local
272 Operator* op = op_it.get(); in ReplaceArrayUsage()
|