Searched refs:GetArgType (Results 1 – 3 of 3) sorted by relevance
294 Status GrapplerFunctionItemInstantiation::GetArgType( in GetArgType() function in tensorflow::grappler::GrapplerFunctionItemInstantiation437 TF_RETURN_IF_ERROR(instantiation.GetArgType(arg, &data_type)); in InstantiationTypeParameters()545 TF_RETURN_IF_ERROR(instantiation.GetArgType(input, &input_data_type)); in MakeGrapplerFunctionItem()608 TF_RETURN_IF_ERROR(instantiation.GetArgType(out, &output_data_type)); in MakeGrapplerFunctionItem()
149 Status GetArgType(const OpDef::ArgDef& arg, DataType* data_type) const;
1137 GetArgType(field) + ") {\n"; in GenStruct()1277 GenTypeAnnotation(kParam, GetArgType(field), GetArgName(field)); in GenStruct()1299 code += ", " + GetArgName(field) + ":" + GetArgType(field); in GenStruct()1342 std::string GetArgType(const FieldDef &field) { in GetArgType() function in flatbuffers::jsts::JsTsGenerator