Searched refs:input_attr (Results 1 – 2 of 2) sorted by relevance
638 input_attr = attr_value_pb2.AttrValue()639 input_attr.list.i.extend(hostmem)640 ret[0].op._set_attr("_input_hostmem", input_attr) # pylint: disable=protected-access768 input_attr = attr_value_pb2.AttrValue()769 input_attr.list.i.extend([num_for_params + i for i in hostmem])770 ret[0].op._set_attr("_input_hostmem", input_attr) # pylint: disable=protected-access
551 const auto input_attr = params_->input_alloc_attrs == nullptr in forward_input() local554 if (!output_attr.IsEqualOrLessRestrictiveThan(input_attr)) { in forward_input()