Searched refs:op_infos (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_executor.cc | 346 SmallVector<OpAsmParser::OperandType, 4> op_infos; in ParseIslandOp() local
347 if (parser.parseOperandList(op_infos, OpAsmParser::Delimiter::OptionalParen)) in ParseIslandOp()
349 if (!op_infos.empty()) { in ParseIslandOp()
350 SmallVector<Type, 2> types(op_infos.size(), control_type); in ParseIslandOp()
351 parser.resolveOperands(op_infos, types, loc, result.operands); in ParseIslandOp()
399 SmallVector<OpAsmParser::OperandType, 2> op_infos; in ParseSwitchOp() local
401 if (parser.parseOperandList(op_infos) || parser.parseColonTypeList(types)) in ParseSwitchOp()
419 if (op_infos.size() < 2) in ParseSwitchOp()
428 types.append(op_infos.size() - 2, control_type); in ParseSwitchOp()
432 if (parser.resolveOperands(op_infos, types, loc, result.operands)) in ParseSwitchOp()
[all …]
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | resource_operation_table.cc | 124 const absl::flat_hash_map<absl::string_view, XlaResourceOpInfo>& op_infos = in GetResourceOpInfoForOp() local
126 auto it = op_infos.find(op); in GetResourceOpInfoForOp()
127 return it == op_infos.end() ? nullptr : &it->second; in GetResourceOpInfoForOp()
|