Searched refs:op_pair (Results 1 – 2 of 2) sorted by relevance
374 for (const auto& op_pair : *op_tape) { in PrepareBackprop()375 op_pair.second.backward_function_deleter( in PrepareBackprop()376 op_pair.second.backward_function); in PrepareBackprop()
1949 auto op_pair = ops.find(node_def.op()); in BinaryTensorOpTensor() local1950 if (op_pair == ops.end()) { in BinaryTensorOpTensor()1995 *const_cast<nvinfer1::ITensor*>(tensor_r), op_pair->second); in BinaryTensorOpTensor()2788 auto op_pair = ops.find(node_def.op()); in ConvertActivation() local2789 if (op_pair == ops.end()) { in ConvertActivation()2799 *const_cast<nvinfer1::ITensor*>(tensor), op_pair->second); in ConvertActivation()3267 auto op_pair = UnaryOperationMap()->find(node_def.op()); in ConvertUnary() local3268 if (op_pair == UnaryOperationMap()->end()) { in ConvertUnary()3277 *const_cast<nvinfer1::ITensor*>(tensor), op_pair->second); in ConvertUnary()