Searched refs:input_def (Results 1 – 2 of 2) sorted by relevance
239 const auto& input_def = ictx->op_def->input_arg(ictx->input_arg_idx++); in OpInferSingleInputAttrs() local240 if (!input_def.number_attr().empty() || !input_def.type_list_attr().empty()) { in OpInferSingleInputAttrs()250 const std::string& type_attr = input_def.type_attr(); in OpInferSingleInputAttrs()259 const tensorflow::OpDef::ArgDef& input_def, in OpInferSingleTypeInputListAttrs() argument263 if (ictx->attrs.find(input_def.number_attr()) == ictx->attrs.end()) { in OpInferSingleTypeInputListAttrs()264 op->operation.MutableAttrs()->Set(input_def.number_attr(), num_inputs); in OpInferSingleTypeInputListAttrs()265 ictx->attrs.insert(input_def.number_attr()); in OpInferSingleTypeInputListAttrs()267 if (ictx->attrs.find(input_def.type_attr()) == ictx->attrs.end()) { in OpInferSingleTypeInputListAttrs()268 op->operation.MutableAttrs()->Set(input_def.type_attr(), in OpInferSingleTypeInputListAttrs()270 ictx->attrs.insert(input_def.type_attr()); in OpInferSingleTypeInputListAttrs()[all …]
285 ArgumentSpec CreateInput(const OpDef_ArgDef& input_def, in CreateInput() argument289 Type type = type_resolver->TypeOf(input_def, &iterable); in CreateInput()