Home
last modified time | relevance | path

Searched refs:op_pair (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/c/eager/
Dtape.h574 for (const auto& op_pair : *op_tape) { in PrepareBackprop()
575 op_pair.second.backward_function_deleter( in PrepareBackprop()
576 op_pair.second.backward_function); in PrepareBackprop()
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes.cc3561 auto op_pair = ActivationTypeMap()->find(activation_mode); in ConvertFusedConv2DBiasActivation() local
3562 if (op_pair == ActivationTypeMap()->end() && activation_mode != "None") { in ConvertFusedConv2DBiasActivation()
3655 if (op_pair != ActivationTypeMap()->end()) { in ConvertFusedConv2DBiasActivation()
3658 op_pair->second); in ConvertFusedConv2DBiasActivation()
3907 auto op_pair = ActivationTypeMap()->find(node_def.op()); in ConvertActivation() local
3908 if (op_pair == ActivationTypeMap()->end()) { in ConvertActivation()
3917 op_pair->second); in ConvertActivation()
4470 auto op_pair = BinaryOperationMap()->find(node_def.op()); in ConvertBinary() local
4471 if (op_pair == BinaryOperationMap()->end()) { in ConvertBinary()
4493 *tensor_l, *tensor_r, op_pair->second); in ConvertBinary()
[all …]