Searched refs:arg_dtypes (Results 1 – 1 of 1) sorted by relevance
1943 const std::vector<DataType>& arg_dtypes, in RemoveArgsToLiftFromFunctionBody() argument1955 arg_node->AddAttr("T", arg_dtypes[i]); in RemoveArgsToLiftFromFunctionBody()1972 ph_builder.Attr("dtype", arg_dtypes[i]); in RemoveArgsToLiftFromFunctionBody()